Update app.py
Browse files
app.py
CHANGED
@@ -261,7 +261,7 @@ def generate_video_from_text_90(
|
|
261 |
"media_items": None,
|
262 |
}
|
263 |
|
264 |
-
generator = torch.Generator(device="
|
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="
|
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)
|