Spaces:
Runtime error
Runtime error
Commit
·
10d711e
1
Parent(s):
dfd39cc
Update app.py
Browse files
app.py
CHANGED
@@ -20,7 +20,7 @@ queue = []
|
|
20 |
# Use the Euler scheduler here instead
|
21 |
scheduler = EulerDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
|
22 |
pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, torch_dtype=torch.float16)
|
23 |
-
pipe = pipe.to("
|
24 |
|
25 |
|
26 |
|
|
|
20 |
# Use the Euler scheduler here instead
|
21 |
scheduler = EulerDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
|
22 |
pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, torch_dtype=torch.float16)
|
23 |
+
pipe = pipe.to("cpu")
|
24 |
|
25 |
|
26 |
|