Spaces:
Running
on
Zero
Running
on
Zero
IceClear
commited on
Commit
·
61e51d7
1
Parent(s):
8207b45
update
Browse files
app.py
CHANGED
|
@@ -387,6 +387,7 @@ def generation_loop(video_path='./test_videos', seed=666, batch_size=1, cfg_scal
|
|
| 387 |
|
| 388 |
gc.collect()
|
| 389 |
torch.cuda.empty_cache()
|
|
|
|
| 390 |
return sample, output_dir
|
| 391 |
|
| 392 |
|
|
|
|
| 387 |
|
| 388 |
gc.collect()
|
| 389 |
torch.cuda.empty_cache()
|
| 390 |
+
print(f"Generated video size: {sample.size()}")
|
| 391 |
return sample, output_dir
|
| 392 |
|
| 393 |
|