Spaces:
Runtime error
Runtime error
Update app.py
Browse fileschange cuda to cpu
app.py
CHANGED
@@ -32,7 +32,7 @@ def generate_image(prompt):
|
|
32 |
prompt,
|
33 |
num_inference_steps=1,
|
34 |
guidance_scale=0,
|
35 |
-
generator=torch.Generator(device="
|
36 |
)
|
37 |
return results.images[0]
|
38 |
|
|
|
32 |
prompt,
|
33 |
num_inference_steps=1,
|
34 |
guidance_scale=0,
|
35 |
+
generator=torch.Generator(device="cpu").manual_seed(seed)
|
36 |
)
|
37 |
return results.images[0]
|
38 |
|