From 34d446cb9829835cf6b8f8fdb4abd9fef3455c3a Mon Sep 17 00:00:00 2001
From: Shkarupa Alex <shkarupa.alex@gmail.com>
Date: Wed, 18 Jun 2025 14:01:25 +0300
Subject: [PATCH] Unify temperature options for Vlm models

---
 docling/datamodel/pipeline_options_vlm_model.py | 2 +-
 docling/models/api_vlm_model.py                 | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/docling/datamodel/pipeline_options_vlm_model.py b/docling/datamodel/pipeline_options_vlm_model.py
index 90ab6685..96c61e86 100644
--- a/docling/datamodel/pipeline_options_vlm_model.py
+++ b/docling/datamodel/pipeline_options_vlm_model.py
@@ -12,6 +12,7 @@ class BaseVlmOptions(BaseModel):
     prompt: str
     scale: float = 2.0
     max_size: Optional[int] = None
+    temperature: float = 0.0
 
 
 class ResponseFormat(str, Enum):
@@ -51,7 +52,6 @@ class InlineVlmOptions(BaseVlmOptions):
         AcceleratorDevice.MPS,
     ]
 
-    temperature: float = 0.0
     stop_strings: List[str] = []
     extra_generation_config: Dict[str, Any] = {}
 
diff --git a/docling/models/api_vlm_model.py b/docling/models/api_vlm_model.py
index bfd00003..63d64a25 100644
--- a/docling/models/api_vlm_model.py
+++ b/docling/models/api_vlm_model.py
@@ -34,7 +34,7 @@ class ApiVlmModel(BasePageModel):
             )
             self.params = {
                 **self.vlm_options.params,
-                "temperature": 0,
+                "temperature": self.vlm_options.temperature,
             }
 
     def __call__(