oxkitsune commited on
Commit
67f5a02
·
1 Parent(s): 043e9ac

don't clear cuda cache

Browse files
Files changed (1) hide show
  1. app.py +0 -1
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