fixed the circular dependenciea

Signed-off-by: Peter Staar <taa@zurich.ibm.com>
This commit is contained in:
Peter Staar 2025-07-10 10:35:47 +02:00
parent 70872e6539
commit b2336830eb
5 changed files with 15 additions and 10 deletions

View File

@ -12,7 +12,7 @@ from docling.datamodel.pipeline_options_asr_model import (
InferenceAsrFramework,
InlineAsrNativeWhisperOptions,
InlineAsrOptions,
TransformersModelType,
# TransformersModelType,
)
_log = logging.getLogger(__name__)

View File

@ -26,8 +26,6 @@ class LayoutModelConfig(BaseModel):
return self.repo_id.replace("/", "--")
# HuggingFace Layout Models
# Default Docling Layout Model
DOCLING_LAYOUT_V2 = LayoutModelConfig(
name="docling_layout_v2",

View File

@ -278,6 +278,7 @@ class VlmPipelineOptions(PaginatedPipelineOptions):
# If True, text from backend will be used instead of generated text
vlm_options: Union[InlineVlmOptions, ApiVlmOptions] = (
smoldocling_vlm_conversion_options
#SMOLDOCLING_TRANSFORMERS
)
@ -293,6 +294,9 @@ class AsrPipelineOptions(PipelineOptions):
artifacts_path: Optional[Union[Path, str]] = None
class PdfPipelineOptions(PaginatedPipelineOptions):
"""Options for the PDF pipeline."""

View File

@ -5,10 +5,11 @@ from pydantic import AnyUrl, BaseModel
from typing_extensions import deprecated
from docling.datamodel.accelerator_options import AcceleratorDevice
from docling.datamodel.pipeline_options_vlm_model import (
# from docling.datamodel.pipeline_options_vlm_model import (
# InferenceFramework,
TransformersModelType,
)
# TransformersModelType,
# )
class BaseAsrOptions(BaseModel):

View File

@ -6,7 +6,6 @@ from pydantic import AnyUrl, BaseModel
from typing_extensions import deprecated
from docling.datamodel.accelerator_options import AcceleratorDevice
from docling.datamodel.pipeline_options import LayoutOptions
class BaseVlmOptions(BaseModel):
@ -89,9 +88,12 @@ class ApiVlmOptions(BaseVlmOptions):
concurrency: int = 1
response_format: ResponseFormat
from docling.datamodel.layout_model_specs import (
LayoutModelConfig,
)
class TwoStageVlmOptions(BaseModel):
kind: Literal["inline_two_stage_model_options"] = "inline_two_stage_model_options"
vlm_options: InlineVlmOptions
layout_options: LayoutOptions
vlm_options: Union[InlineVlmOptions, ApiVlmOptions] # = SMOLDOCLING_TRANSFORMERS
layout_options: LayoutModelConfig # = DOCLING_LAYOUT_V2