Spaces:
Sleeping
Sleeping
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):
|