Spaces:
Runtime error
Runtime error
cocktailpeanut
commited on
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")
|