mirrors/docling
1
0
Fork 0
mirror of https://github.com/DS4SD/docling.git synced 2025-07-27 04:24:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

updated comment

Browse Source 
Signed-off-by: Peter Staar <taa@zurich.ibm.com>
This commit is contained in:
Peter Staar 2025-06-20 05:06:37 +02:00
parent 69b271d09c
commit 408b03ebbc
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
docling/datamodel/pipeline_options.py
View File

@ -11,9 +11,10 @@ from pydantic import (
) )
from typing_extensions import deprecated from typing_extensions import deprecated
from docling.datamodel import asr_model_specs
# Import the following for backwards compatibility # Import the following for backwards compatibility
from docling.datamodel.accelerator_options import AcceleratorDevice, AcceleratorOptions from docling.datamodel.accelerator_options import AcceleratorDevice, AcceleratorOptions
from docling.datamodel import asr_model_specs
from docling.datamodel.pipeline_options_asr_model import ( from docling.datamodel.pipeline_options_asr_model import (
InlineAsrOptions, InlineAsrOptions,
) )

2
docling/pipeline/asr_pipeline.py
View File

@ -113,7 +113,7 @@ class _NativeWhisperModel:
import whisper # type: ignore import whisper # type: ignore
except ImportError: except ImportError:
raise ImportError( raise ImportError(
"whisper is not installed. Please install it via `pip install openai-whisper`." "whisper is not installed. Please install it via `pip install openai-whisper` or do `uv sync --extra asr`."
) )
self.asr_options = asr_options self.asr_options = asr_options
self.max_tokens = asr_options.max_new_tokens self.max_tokens = asr_options.max_new_tokens