Spaces:
Running
on
Zero
Running
on
Zero
limit batch size even more
Browse files
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(
|
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):
|