ford442 commited on
Commit
3b168f4
·
verified ·
1 Parent(s): ee4cc69

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -254,7 +254,7 @@ def generate_video_from_text(
254
  "media_items": None,
255
  }
256
 
257
- generator = torch.Generator(device="cpu").manual_seed(seed)
258
 
259
  def gradio_progress_callback(self, step, timestep, kwargs):
260
  progress((step + 1) / num_inference_steps)
 
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)