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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -193,8 +193,6 @@ def preset_changed(preset):
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)
200
  unet = load_unet(unet_dir)
@@ -258,6 +256,7 @@ def generate_video_from_text(
258
  "media_items": None,
259
  }
260
 
 
261
 
262
  def gradio_progress_callback(self, step, timestep, kwargs):
263
  progress((step + 1) / num_inference_steps)
 
193
  gr.update(visible=True),
194
  )
195
 
 
 
196
  # Load models
197
  vae = load_vae(vae_dir)
198
  unet = load_unet(unet_dir)
 
256
  "media_items": None,
257
  }
258
 
259
+ generator = torch.Generator(device="cpu").manual_seed(seed)
260
 
261
  def gradio_progress_callback(self, step, timestep, kwargs):
262
  progress((step + 1) / num_inference_steps)