mirrors/docling
1
0
Fork 0
mirror of https://github.com/DS4SD/docling.git synced 2025-07-31 14:34:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: Accept AcceleratorDevice enum type

Browse Source 
Signed-off-by: Christoph Auer <cau@zurich.ibm.com>
This commit is contained in:
Christoph Auer 2025-02-03 14:32:44 +01:00 committed by ahn
parent dd0728e646
commit 0bca78a84b
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docling/datamodel/pipeline_options.py
View File

@ -39,7 +39,7 @@ class AcceleratorOptions(BaseSettings):
)
num_threads: int = 4
device: str = "auto"
device: Union[str, AcceleratorDevice] = "auto"
@field_validator("device")
def validate_device(cls, value):