fastvideogen

Running on Zero

ford442 commited on Dec 2, 2024

Commit

d5eb5a1

verified ·

1 Parent(s): f17c1bf

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -256,7 +256,7 @@ def generate_video_from_text(
         "media_items": None,
     }
-    generator = torch.Generator(device="cpu").manual_seed(seed)
     def gradio_progress_callback(self, step, timestep, kwargs):
         progress((step + 1) / num_inference_steps)
@@ -352,7 +352,7 @@ def generate_video_from_image(
         "media_items": media_items,
     }
-    generator = torch.Generator(device="cpu").manual_seed(seed)
     def gradio_progress_callback(self, step, timestep, kwargs):
         progress((step + 1) / num_inference_steps)

         "media_items": None,
     }
+    generator = torch.Generator(device="cuda").manual_seed(seed)
     def gradio_progress_callback(self, step, timestep, kwargs):
         progress((step + 1) / num_inference_steps)
         "media_items": media_items,
     }
+    generator = torch.Generator(device="cuda").manual_seed(seed)
     def gradio_progress_callback(self, step, timestep, kwargs):
         progress((step + 1) / num_inference_steps)