logit_bias field added to the completion request (#14)

This commit is contained in:
Viacheslav Kovalevskyi
2022-01-10 10:05:23 +03:00
committed by GitHub
parent 7ff9fedf93
commit eecd212963
+4 -3
View File
@@ -25,9 +25,10 @@ type CompletionRequest struct {
Echo bool `json:"echo,omitempty"` Echo bool `json:"echo,omitempty"`
Stop []string `json:"stop,omitempty"` Stop []string `json:"stop,omitempty"`
PresencePenalty float32 `json:"presence_penalty,omitempty"` PresencePenalty float32 `json:"presence_penalty,omitempty"`
FrequencyPenalty float32 `json:"frequency_penalty,omitempty"` FrequencyPenalty float32 `json:"frequency_penalty,omitempty"`
BestOf int `json:"best_of,omitempty"` BestOf int `json:"best_of,omitempty"`
LogitBias map[string]int `json:"logit_bias,omitempty"`
} }
// Choice represents one of possible completions // Choice represents one of possible completions