Update app.py
Browse files
app.py
CHANGED
|
@@ -13,7 +13,7 @@ device = "cuda"
|
|
| 13 |
pipe = DiffusionPipeline.from_pretrained(
|
| 14 |
"CompVis/stable-diffusion-v1-4",
|
| 15 |
safety_checker=None,
|
| 16 |
-
custom_pipeline=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 |
|
|
|
|
| 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 |
|