From 0bca78a84b2aa8c1b15f413820e15da2c8c33d88 Mon Sep 17 00:00:00 2001
From: Christoph Auer <cau@zurich.ibm.com>
Date: Mon, 3 Feb 2025 14:32:44 +0100
Subject: [PATCH] chore: Accept AcceleratorDevice enum type

Signed-off-by: Christoph Auer <cau@zurich.ibm.com>
---
 docling/datamodel/pipeline_options.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/docling/datamodel/pipeline_options.py b/docling/datamodel/pipeline_options.py
index aec6a5ac..c1721964 100644
--- a/docling/datamodel/pipeline_options.py
+++ b/docling/datamodel/pipeline_options.py
@@ -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):