Spaces:
Runtime error
Runtime error
Commit
•
233676c
1
Parent(s):
ee3d220
Update app.py
Browse files
app.py
CHANGED
|
@@ -140,7 +140,7 @@ def inference(model_name, prompt, guidance, steps, n_images=1, width=512, height
|
|
| 140 |
seed = random.randint(0, 2147483647)
|
| 141 |
|
| 142 |
if torch.cuda.is_available():
|
| 143 |
-
|
| 144 |
else:
|
| 145 |
generator = torch.Generator().manual_seed(seed)
|
| 146 |
|
|
|
|
| 140 |
seed = random.randint(0, 2147483647)
|
| 141 |
|
| 142 |
if torch.cuda.is_available():
|
| 143 |
+
generator = torch.Generator('cuda').manual_seed(seed)
|
| 144 |
else:
|
| 145 |
generator = torch.Generator().manual_seed(seed)
|
| 146 |
|