fffiloni commited on
Commit
bcc70ee
1 Parent(s): 4505814

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,7 +3,7 @@ from PIL import Image
3
  from io import BytesIO
4
  import torch
5
  import os
6
- import imagic
7
  from diffusers import DiffusionPipeline, DDIMScheduler, UNet2DConditionModel
8
  MY_SECRET_TOKEN=os.environ.get('HF_TOKEN_SD')
9
 
@@ -13,7 +13,7 @@ device = "cuda"
13
  pipe = DiffusionPipeline.from_pretrained(
14
  "CompVis/stable-diffusion-v1-4",
15
  safety_checker=None,
16
- custom_pipeline="https://huggingface.co/spaces/fffiloni/imagic-stable-diffusion/blob/main/imagic.py",
17
  scheduler = DDIMScheduler(beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", clip_sample=False, set_alpha_to_one=False)
18
  ).to(device)
19
 
 
3
  from io import BytesIO
4
  import torch
5
  import os
6
+
7
  from diffusers import DiffusionPipeline, DDIMScheduler, UNet2DConditionModel
8
  MY_SECRET_TOKEN=os.environ.get('HF_TOKEN_SD')
9
 
 
13
  pipe = DiffusionPipeline.from_pretrained(
14
  "CompVis/stable-diffusion-v1-4",
15
  safety_checker=None,
16
+ custom_pipeline="imagic_sylvain",
17
  scheduler = DDIMScheduler(beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", clip_sample=False, set_alpha_to_one=False)
18
  ).to(device)
19