rename to specs

Signed-off-by: Michele Dolfi <dol@zurich.ibm.com>
This commit is contained in:
Michele Dolfi 2025-06-02 10:40:06 +02:00
parent 76718cb1f9
commit b9c1698263
6 changed files with 12 additions and 8 deletions

View File

@ -47,7 +47,7 @@ from docling.datamodel.pipeline_options import (
VlmPipelineOptions,
)
from docling.datamodel.settings import settings
from docling.datamodel.vlm_model_spec import (
from docling.datamodel.vlm_model_specs import (
GRANITE_VISION_OLLAMA,
GRANITE_VISION_TRANSFORMERS,
SMOLDOCLING_MLX,

View File

@ -19,7 +19,7 @@ from docling.datamodel.pipeline_options_vlm_model import (
InlineVlmOptions,
ResponseFormat,
)
from docling.datamodel.vlm_model_spec import (
from docling.datamodel.vlm_model_specs import (
GRANITE_VISION_OLLAMA as granite_vision_vlm_ollama_conversion_options,
GRANITE_VISION_TRANSFORMERS as granite_vision_vlm_conversion_options,
SMOLDOCLING_MLX as smoldocling_vlm_mlx_conversion_options,

View File

@ -44,10 +44,14 @@ SMOLDOCLING_TRANSFORMERS = InlineVlmOptions(
# GraniteVision
GRANITE_VISION_TRANSFORMERS = InlineVlmOptions(
repo_id="ibm-granite/granite-vision-3.2-2b",
prompt="Convert this page to markdown. Do not miss any text and only output the bare MarkDown!",
prompt="Convert this page to markdown. Do not miss any text and only output the bare markdown!",
response_format=ResponseFormat.MARKDOWN,
inference_framework=InferenceFramework.TRANSFORMERS_VISION2SEQ,
supported_devices=[AcceleratorDevice.CPU, AcceleratorDevice.CUDA],
supported_devices=[
AcceleratorDevice.CPU,
AcceleratorDevice.CUDA,
AcceleratorDevice.MPS,
],
scale=2.0,
temperature=0.0,
)
@ -55,7 +59,7 @@ GRANITE_VISION_TRANSFORMERS = InlineVlmOptions(
GRANITE_VISION_OLLAMA = ApiVlmOptions(
url=AnyUrl("http://localhost:11434/v1/chat/completions"),
params={"model": "granite3.2-vision:2b"},
prompt="Convert this page to markdown. Do not miss any text and only output the bare MarkDown!",
prompt="Convert this page to markdown. Do not miss any text and only output the bare markdown!",
scale=1.0,
timeout=120,
response_format=ResponseFormat.MARKDOWN,

View File

@ -7,7 +7,7 @@ from docling.datamodel.pipeline_options import (
smolvlm_picture_description,
)
from docling.datamodel.settings import settings
from docling.datamodel.vlm_model_spec import (
from docling.datamodel.vlm_model_specs import (
SMOLDOCLING_MLX,
SMOLDOCLING_TRANSFORMERS,
)

View File

@ -16,7 +16,7 @@ from docling.datamodel.base_models import InputFormat
from docling.datamodel.pipeline_options import (
VlmPipelineOptions,
)
from docling.datamodel.vlm_model_spec import (
from docling.datamodel.vlm_model_specs import (
GEMMA3_12B_MLX,
GRANITE_VISION_OLLAMA,
GRANITE_VISION_TRANSFORMERS,

View File

@ -2,7 +2,7 @@ from docling.datamodel.base_models import InputFormat
from docling.datamodel.pipeline_options import (
VlmPipelineOptions,
)
from docling.datamodel.vlm_model_spec import SMOLDOCLING_MLX
from docling.datamodel.vlm_model_specs import SMOLDOCLING_MLX
from docling.document_converter import DocumentConverter, PdfFormatOption
from docling.pipeline.vlm_pipeline import VlmPipeline