mirrors/docling
1
0
Fork 0
mirror of https://github.com/DS4SD/docling.git synced 2025-08-01 23:12:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

remove batch size from CLI

Browse Source 
Signed-off-by: Michele Dolfi <dol@zurich.ibm.com>
This commit is contained in:
Michele Dolfi 2025-02-05 09:51:55 +01:00
parent 00aac6b620
commit e7b89e7815
2 changed files with 0 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
docling/cli/main.py
View File

@ -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)

2
docling/models/code_formula_model.py
View File

@ -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):