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