ford442 commited on
Commit
e22c28a
·
verified ·
1 Parent(s): fb88786

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -193,7 +193,7 @@ def preset_changed(preset):
193
  gr.update(visible=True),
194
  )
195
 
196
- generator = torch.Generator(device="cuda").manual_seed(seed)
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)