Improve Mistral format checks. (#33253)

Signed-off-by: Julien Denize <julien.denize@mistral.ai>
Signed-off-by: juliendenize <julien.denize@mistral.ai>
Co-authored-by: Michael Goin <mgoin64@gmail.com>
This commit is contained in:
Julien Denize
2026-01-30 15:23:33 +01:00
committed by GitHub
parent a11bc12d53
commit ae5b7aff2b
8 changed files with 193 additions and 24 deletions

View File

@@ -365,6 +365,7 @@ class HfRunner:
self.config,
dtype=dtype,
is_pooling_model=is_sentence_transformer or is_cross_encoder,
config_format="hf",
)
model_kwargs = model_kwargs if model_kwargs is not None else {}