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

Update model.go with max_tokens param

Browse Source
This commit is contained in:
MotorBottle 2024-07-24 20:05:53 +08:00 committed by GitHub
parent 8c40a1dae0
commit 804fd7e405
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/model.go
View File

@ -7,6 +7,7 @@ type Options struct {
TopP float64 `json:"top_p,omitempty"` TopP float64 `json:"top_p,omitempty"`
FrequencyPenalty float64 `json:"frequency_penalty,omitempty"` FrequencyPenalty float64 `json:"frequency_penalty,omitempty"`
PresencePenalty float64 `json:"presence_penalty,omitempty"` PresencePenalty float64 `json:"presence_penalty,omitempty"`
NumPredict float64 `json:"num_predict,omitempty"`
} }
type Message struct { type Message struct {