Ecosystem/ai-gateway
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update main.go with max_tokens param

Browse Source
This commit is contained in:
MotorBottle 2024-07-24 20:04:20 +08:00 committed by GitHub
parent c936198ac8
commit 8c40a1dae0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
relay/adaptor/ollama/main.go
View File

@ -31,6 +31,7 @@ func ConvertRequest(request model.GeneralOpenAIRequest) *ChatRequest {
TopP: request.TopP, TopP: request.TopP,
FrequencyPenalty: request.FrequencyPenalty, FrequencyPenalty: request.FrequencyPenalty,
PresencePenalty: request.PresencePenalty, PresencePenalty: request.PresencePenalty,
NumPredict: request.MaxTokens,
}, },
Stream: request.Stream, Stream: request.Stream,
} }