Spaces:
Runtime error
Runtime error
Commit
·
d7bca4c
1
Parent(s):
43f619a
update
Browse files
app.py
CHANGED
|
@@ -16,7 +16,7 @@ else:
|
|
| 16 |
device = "cpu"
|
| 17 |
model_id = "stabilityai/stable-diffusion-2-1-base"
|
| 18 |
|
| 19 |
-
to = torch.float16 if
|
| 20 |
|
| 21 |
# components for the Preprocessor
|
| 22 |
scheduler = DDIMScheduler.from_pretrained(model_id, subfolder="scheduler")
|
|
|
|
| 16 |
device = "cpu"
|
| 17 |
model_id = "stabilityai/stable-diffusion-2-1-base"
|
| 18 |
|
| 19 |
+
to = torch.float16 if device == 'cuda' else torch.float32
|
| 20 |
|
| 21 |
# components for the Preprocessor
|
| 22 |
scheduler = DDIMScheduler.from_pretrained(model_id, subfolder="scheduler")
|