fix(utils.py): mapping stop sequences for palm

This commit is contained in:
Krrish Dholakia 2023-11-01 14:00:45 -07:00
parent a46d6a2dc9
commit f9ff03d5af
2 changed files with 5 additions and 5 deletions

View file

@ -1421,7 +1421,7 @@ def get_optional_params( # use the openai defaults
if n:
optional_params["candidate_count"] = n
if stop:
optional_params["stopSequences"] = stop
optional_params["stop_sequences"] = stop
if max_tokens:
optional_params["max_output_tokens"] = max_tokens
elif (