Fix page_range stopping at page 32 by using dynamic batch_size

Co-authored-by: cau-git <60343111+cau-git@users.noreply.github.com>
This commit is contained in:
copilot-swe-agent[bot]
2025-11-19 13:25:00 +00:00
parent 18f705b235
commit 58fc6ccf86
2 changed files with 2 additions and 2 deletions

View File

@@ -252,7 +252,7 @@ class ThreadedLayoutVlmPipeline(BasePipeline):
proc = ProcessingResult(total_expected=total_pages)
fed_idx = 0
batch_size = 32
batch_size = total_pages
try:
while proc.success_count + proc.failure_count < total_pages: