Skip to content

Commit 98fde58

Browse files
committed
Map Layout+VLM pipeline to GraniteDoclign
Signed-off-by: Christoph Auer <[email protected]>
1 parent 0a39e2f commit 98fde58

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

docling/experimental/datamodel/threaded_layout_vlm_pipeline_options.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
ApiVlmOptions,
99
InlineVlmOptions,
1010
)
11-
from docling.datamodel.vlm_model_specs import SMOLDOCLING_MLX, SMOLDOCLING_TRANSFORMERS
11+
from docling.datamodel.vlm_model_specs import GRANITEDOCLING_TRANSFORMERS
1212

1313

1414
class ThreadedLayoutVlmPipelineOptions(PaginatedPipelineOptions):
@@ -17,7 +17,7 @@ class ThreadedLayoutVlmPipelineOptions(PaginatedPipelineOptions):
1717
images_scale: float = 2.0
1818

1919
# VLM configuration (will be enhanced with layout awareness by the pipeline)
20-
vlm_options: Union[InlineVlmOptions, ApiVlmOptions] = SMOLDOCLING_TRANSFORMERS
20+
vlm_options: Union[InlineVlmOptions, ApiVlmOptions] = GRANITEDOCLING_TRANSFORMERS
2121

2222
# Layout model configuration
2323
layout_options: LayoutOptions = LayoutOptions(

0 commit comments

Comments
 (0)