Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -254,7 +254,7 @@ def generate_video_from_text(
|
|
254 |
"media_items": None,
|
255 |
}
|
256 |
|
257 |
-
generator = torch.Generator(device="
|
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)
|