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 |
|