Spaces:
Running
on
Zero
Running
on
Zero
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 |
|