Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -193,7 +193,7 @@ def preset_changed(preset):
|
|
193 |
gr.update(visible=True),
|
194 |
)
|
195 |
|
196 |
-
generator = torch.Generator(device="
|
197 |
|
198 |
# Load models
|
199 |
vae = load_vae(vae_dir)
|
|
|
193 |
gr.update(visible=True),
|
194 |
)
|
195 |
|
196 |
+
generator = torch.Generator(device="cpu").manual_seed(seed)
|
197 |
|
198 |
# Load models
|
199 |
vae = load_vae(vae_dir)
|