Spaces:
Running
Running
Commit
·
6e8765b
1
Parent(s):
7913e3a
Update app.py
Browse files
app.py
CHANGED
@@ -15,8 +15,11 @@ from diffusers import StableDiffusionPanoramaPipeline, DDIMScheduler
|
|
15 |
|
16 |
model_ckpt = "stabilityai/stable-diffusion-2-base"
|
17 |
scheduler = DDIMScheduler.from_pretrained(model_ckpt, subfolder="scheduler")
|
|
|
|
|
|
|
18 |
pipe = StableDiffusionPanoramaPipeline.from_pretrained(
|
19 |
-
model_ckpt, scheduler=scheduler
|
20 |
)
|
21 |
|
22 |
pipe = pipe.to(torch.device("cuda" if torch.cuda.is_available() else "cpu"))
|
|
|
15 |
|
16 |
model_ckpt = "stabilityai/stable-diffusion-2-base"
|
17 |
scheduler = DDIMScheduler.from_pretrained(model_ckpt, subfolder="scheduler")
|
18 |
+
# pipe = StableDiffusionPanoramaPipeline.from_pretrained(
|
19 |
+
# model_ckpt, scheduler=scheduler, torch_dtype=torch.float16
|
20 |
+
# )
|
21 |
pipe = StableDiffusionPanoramaPipeline.from_pretrained(
|
22 |
+
model_ckpt, scheduler=scheduler
|
23 |
)
|
24 |
|
25 |
pipe = pipe.to(torch.device("cuda" if torch.cuda.is_available() else "cpu"))
|