mirror of
https://github.com/DS4SD/docling.git
synced 2025-12-08 20:58:11 +00:00
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:
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user