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