ford442 commited on
Commit
9bdea9f
1 Parent(s): b87dd23

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -261,7 +261,7 @@ def generate_video_from_text_90(
261
  "media_items": None,
262
  }
263
 
264
- generator = torch.Generator(device="cpu").manual_seed(seed)
265
 
266
  def gradio_progress_callback(self, step, timestep, kwargs):
267
  progress((step + 1) / num_inference_steps)
@@ -356,7 +356,7 @@ def generate_video_from_image_90(
356
  "media_items": media_items,
357
  }
358
 
359
- generator = torch.Generator(device="cpu").manual_seed(seed)
360
 
361
  def gradio_progress_callback(self, step, timestep, kwargs):
362
  progress((step + 1) / num_inference_steps)
 
261
  "media_items": None,
262
  }
263
 
264
+ generator = torch.Generator(device="cuda").manual_seed(seed)
265
 
266
  def gradio_progress_callback(self, step, timestep, kwargs):
267
  progress((step + 1) / num_inference_steps)
 
356
  "media_items": media_items,
357
  }
358
 
359
+ generator = torch.Generator(device="cuda").manual_seed(seed)
360
 
361
  def gradio_progress_callback(self, step, timestep, kwargs):
362
  progress((step + 1) / num_inference_steps)