Update app.py
Browse files
app.py
CHANGED
@@ -12,8 +12,6 @@ device = "cuda"
|
|
12 |
pipe = DiffusionPipeline.from_pretrained(
|
13 |
"CompVis/stable-diffusion-v1-4",
|
14 |
safety_checker=None,
|
15 |
-
use_auth_token=MY_SECRET_TOKEN,
|
16 |
-
revision="fp16",
|
17 |
custom_pipeline="imagic_stable_diffusion",
|
18 |
scheduler = DDIMScheduler(beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", clip_sample=False, set_alpha_to_one=False)
|
19 |
).to(device)
|
|
|
12 |
pipe = DiffusionPipeline.from_pretrained(
|
13 |
"CompVis/stable-diffusion-v1-4",
|
14 |
safety_checker=None,
|
|
|
|
|
15 |
custom_pipeline="imagic_stable_diffusion",
|
16 |
scheduler = DDIMScheduler(beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", clip_sample=False, set_alpha_to_one=False)
|
17 |
).to(device)
|