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

limit batch size even more

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -111,7 +111,7 @@ def run_rerun(path_to_video):
111
  free_vram = free_vram / 1024 / 1024 / 1024
112
 
113
  # batch size is determined by the amount of free vram
114
- batch_size = int(min(min(16, free_vram // 4), max_frames))
115
 
116
  # go through all the frames in the video, using the batch size
117
  for i in range(0, int(max_frames), batch_size):
 
111
  free_vram = free_vram / 1024 / 1024 / 1024
112
 
113
  # batch size is determined by the amount of free vram
114
+ batch_size = int(min(min(8, free_vram // 4), max_frames))
115
 
116
  # go through all the frames in the video, using the batch size
117
  for i in range(0, int(max_frames), batch_size):