fix _supports_penalty_parameters

This commit is contained in:
Ishaan Jaffer 2025-10-17 17:57:24 -07:00
parent 9c6de6972f
commit b19d9435cf

View file

@ -691,11 +691,9 @@ class VertexGeminiConfig(VertexAIBaseConfig, BaseConfig):
value=value, optional_params=optional_params
)
elif param == "frequency_penalty":
optional_params["frequency_penalty"] = value
if self._supports_penalty_parameters(model):
optional_params["frequency_penalty"] = value
elif param == "presence_penalty":
optional_params["presence_penalty"] = value
if self._supports_penalty_parameters(model):
optional_params["presence_penalty"] = value
elif param == "logprobs":