From e7b89e7815aeee071d32d560bbcc10ea76b287c3 Mon Sep 17 00:00:00 2001 From: Michele Dolfi Date: Wed, 5 Feb 2025 09:51:55 +0100 Subject: [PATCH] remove batch size from CLI Signed-off-by: Michele Dolfi --- docling/cli/main.py | 6 ------ docling/models/code_formula_model.py | 2 -- 2 files changed, 8 deletions(-) diff --git a/docling/cli/main.py b/docling/cli/main.py index 2ca28866..4a775cb9 100644 --- a/docling/cli/main.py +++ b/docling/cli/main.py @@ -290,12 +290,6 @@ def convert( device: Annotated[ AcceleratorDevice, typer.Option(..., help="Accelerator device") ] = AcceleratorDevice.AUTO, - batch_size_code_formula: Annotated[ - int, typer.Option(..., help="Batch size for the code and formula model") - ] = CodeFormulaModel.elements_batch_size, - batch_size_picture_classes: Annotated[ - int, typer.Option(..., help="Batch size for the picture classifier") - ] = DocumentPictureClassifier.elements_batch_size, ): if verbose == 0: logging.basicConfig(level=logging.WARNING) diff --git a/docling/models/code_formula_model.py b/docling/models/code_formula_model.py index d3752d7f..6648f465 100644 --- a/docling/models/code_formula_model.py +++ b/docling/models/code_formula_model.py @@ -235,8 +235,6 @@ class CodeFormulaModel(BaseItemAndImageEnrichmentModel): labels.append(el.item.label) images.append(el.image) - print(f"BATCH: {len(images)=}") - outputs = self.code_formula_model.predict(images, labels) for item, output in zip(elements, outputs):