diff --git a/relay/adaptor/ollama/model.go b/relay/adaptor/ollama/model.go index 94947ec7..8ed4174e 100644 --- a/relay/adaptor/ollama/model.go +++ b/relay/adaptor/ollama/model.go @@ -7,7 +7,7 @@ type Options struct { TopP float64 `json:"top_p,omitempty"` FrequencyPenalty float64 `json:"frequency_penalty,omitempty"` PresencePenalty float64 `json:"presence_penalty,omitempty"` - NumPredict float64 `json:"num_predict,omitempty"` + NumPredict int `json:"num_predict,omitempty"` } type Message struct {