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)
|