Update inference/flovd_demo.py
Browse files- inference/flovd_demo.py +1 -1
inference/flovd_demo.py
CHANGED
|
@@ -303,7 +303,7 @@ def generate_video(
|
|
| 303 |
- seed (int): The seed for reproducibility.
|
| 304 |
- fps (int): The frames per second for the generated video.
|
| 305 |
"""
|
| 306 |
-
|
| 307 |
local_rank = 'cuda'
|
| 308 |
|
| 309 |
torch.manual_seed(seed)
|
|
|
|
| 303 |
- seed (int): The seed for reproducibility.
|
| 304 |
- fps (int): The frames per second for the generated video.
|
| 305 |
"""
|
| 306 |
+
print("at generate video", flush=True)
|
| 307 |
local_rank = 'cuda'
|
| 308 |
|
| 309 |
torch.manual_seed(seed)
|