update

app.py

@@ -387,6 +387,7 @@ def generation_loop(video_path='./test_videos', seed=666, batch_size=1, cfg_scal
 
         gc.collect()
         torch.cuda.empty_cache()
+        print(f"Generated video size: {sample.size()}")
         return sample, output_dir