oxkitsune commited on
Commit
906a232
·
1 Parent(s): 67f5a02

even smaller batch size?

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