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