Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -123,7 +123,7 @@ def inference(model_name, prompt, guidance, steps, n_images=1, width=512, height
|
|
| 123 |
if seed == 0:
|
| 124 |
seed = random.randint(0, 2147483647)
|
| 125 |
|
| 126 |
-
generator = torch.Generator('
|
| 127 |
|
| 128 |
try:
|
| 129 |
if img is not None:
|
|
|
|
| 123 |
if seed == 0:
|
| 124 |
seed = random.randint(0, 2147483647)
|
| 125 |
|
| 126 |
+
generator = torch.Generator('cpu').manual_seed(seed)
|
| 127 |
|
| 128 |
try:
|
| 129 |
if img is not None:
|