mirror of
https://github.com/DS4SD/docling.git
synced 2025-07-23 18:45:00 +00:00
Unify temperature options for Vlm models
This commit is contained in:
parent
215b540f6c
commit
34d446cb98
@ -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] = {}
|
||||
|
||||
|
@ -34,7 +34,7 @@ class ApiVlmModel(BasePageModel):
|
||||
)
|
||||
self.params = {
|
||||
**self.vlm_options.params,
|
||||
"temperature": 0,
|
||||
"temperature": self.vlm_options.temperature,
|
||||
}
|
||||
|
||||
def __call__(
|
||||
|
Loading…
Reference in New Issue
Block a user