Spaces:
Running
Running
don't clear cuda cache
Browse files
app.py
CHANGED
|
@@ -106,7 +106,6 @@ def run_rerun(path_to_video):
|
|
| 106 |
# limit the number of frames to 10 seconds of video
|
| 107 |
max_frames = min(10 * fps_video, num_frames)
|
| 108 |
|
| 109 |
-
torch.cuda.empty_cache()
|
| 110 |
free_vram, _ = torch.cuda.mem_get_info(device)
|
| 111 |
free_vram = free_vram / 1024 / 1024 / 1024
|
| 112 |
|
|
|
|
| 106 |
# limit the number of frames to 10 seconds of video
|
| 107 |
max_frames = min(10 * fps_video, num_frames)
|
| 108 |
|
|
|
|
| 109 |
free_vram, _ = torch.cuda.mem_get_info(device)
|
| 110 |
free_vram = free_vram / 1024 / 1024 / 1024
|
| 111 |
|