Fixing Max Tokens
This commit is contained in:
parent
6966d575d5
commit
6d99028d91
@ -168,7 +168,7 @@ class Pipeline:
|
|||||||
payload = {"modelId": model_id,
|
payload = {"modelId": model_id,
|
||||||
"messages": processed_messages,
|
"messages": processed_messages,
|
||||||
"system": [{'text': system_message if system_message else 'you are an intelligent ai assistant'}],
|
"system": [{'text': system_message if system_message else 'you are an intelligent ai assistant'}],
|
||||||
"inferenceConfig": {"temperature": 1, "max_tokens": body.get("max_tokens", MAX_COMBINED_TOKENS)},
|
"inferenceConfig": {"temperature": 1, "maxTokens": body.get("max_tokens", MAX_COMBINED_TOKENS)},
|
||||||
"additionalModelRequestFields": reasoning_config
|
"additionalModelRequestFields": reasoning_config
|
||||||
}
|
}
|
||||||
if body.get("stream", False):
|
if body.get("stream", False):
|
||||||
|
Loading…
x
Reference in New Issue
Block a user