Spaces:
Running
on
Zero
Running
on
Zero
IceClear
commited on
Commit
·
7c41897
1
Parent(s):
e276ced
update
Browse files
app.py
CHANGED
|
@@ -386,7 +386,6 @@ def generation_loop(video_path='./test_videos', output_dir='./results', seed=666
|
|
| 386 |
sample = sample.to(torch.uint8).numpy()
|
| 387 |
|
| 388 |
gc.collect()
|
| 389 |
-
torch.cuda.empty_cache()
|
| 390 |
return sample, output_dir
|
| 391 |
|
| 392 |
|
|
|
|
| 386 |
sample = sample.to(torch.uint8).numpy()
|
| 387 |
|
| 388 |
gc.collect()
|
|
|
|
| 389 |
return sample, output_dir
|
| 390 |
|
| 391 |
|