ford442 commited on
Commit
0ca91b3
·
verified ·
1 Parent(s): fe6bb2b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -356,7 +356,7 @@ def generate_video_from_image_90(
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)
 
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)