ford442 commited on
Commit
1a17be8
1 Parent(s): 3b168f4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -191,6 +191,8 @@ def preset_changed(preset):
191
  gr.update(visible=True),
192
  gr.update(visible=True),
193
  )
 
 
194
 
195
  # Load models
196
  vae = load_vae(vae_dir)
@@ -254,7 +256,6 @@ def generate_video_from_text(
254
  "media_items": None,
255
  }
256
 
257
- generator = torch.Generator(device="cuda").manual_seed(seed)
258
 
259
  def gradio_progress_callback(self, step, timestep, kwargs):
260
  progress((step + 1) / num_inference_steps)
 
191
  gr.update(visible=True),
192
  gr.update(visible=True),
193
  )
194
+
195
+ generator = torch.Generator(device="cuda").manual_seed(seed)
196
 
197
  # Load models
198
  vae = load_vae(vae_dir)
 
256
  "media_items": None,
257
  }
258
 
 
259
 
260
  def gradio_progress_callback(self, step, timestep, kwargs):
261
  progress((step + 1) / num_inference_steps)