Spaces:
Running
Running
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)
|