logit_bias field added to the completion request (#14)

This commit is contained in:
Viacheslav Kovalevskyi
2022-01-09 23:05:23 -08:00
committed by GitHub
parent 7ff9fedf93
commit eecd212963

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