Skip to content

Commit cbae57a

Browse files
committed
fix(extract-python): add docling-gpu extra to support ONNX GPU and nemotron
1 parent 47326ef commit cbae57a

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

‎extract-python/pyproject.toml‎

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,11 @@ benches = [
2222
]
2323

2424
docling = [
25-
"docling-slim[standard,feat-ocr-easyocr,feat-ocr-tesserocr,feat-ocr-rapidocr-onnx,feat-ocr-mac,feat-ocr-nemotron]~=2.127",
25+
"docling-slim[standard,feat-ocr-easyocr,feat-ocr-tesserocr,feat-ocr-rapidocr,feat-ocr-mac]~=2.127",
26+
]
27+
28+
docling-gpu = [
29+
"docling-slim[feat-ocr-nemotron,models-onnxruntime,models-vlm-inline]~=2.127",
2630
]
2731

2832
marker = [

0 commit comments

Comments
 (0)