mirror of
https://github.com/mudler/LocalAI.git
synced 2024-06-07 19:40:48 +00:00
801b481beb
* fix request debugging, disable marshalling of context fields Signed-off-by: blob42 <contact@blob42.xyz> * merge frequency_penalty request parm with config Signed-off-by: blob42 <contact@blob42.xyz> * openai: add presence_penalty parameter Signed-off-by: blob42 <contact@blob42.xyz> --------- Signed-off-by: blob42 <contact@blob42.xyz>
47 lines
1.7 KiB
Go
47 lines
1.7 KiB
Go
package schema
|
|
|
|
type PredictionOptions struct {
|
|
|
|
// Also part of the OpenAI official spec
|
|
Model string `json:"model" yaml:"model"`
|
|
|
|
// Also part of the OpenAI official spec
|
|
Language string `json:"language"`
|
|
|
|
// Also part of the OpenAI official spec. use it for returning multiple results
|
|
N int `json:"n"`
|
|
|
|
// Common options between all the API calls, part of the OpenAI spec
|
|
TopP *float64 `json:"top_p" yaml:"top_p"`
|
|
TopK *int `json:"top_k" yaml:"top_k"`
|
|
Temperature *float64 `json:"temperature" yaml:"temperature"`
|
|
Maxtokens *int `json:"max_tokens" yaml:"max_tokens"`
|
|
Echo bool `json:"echo"`
|
|
|
|
// Custom parameters - not present in the OpenAI API
|
|
Batch int `json:"batch" yaml:"batch"`
|
|
IgnoreEOS bool `json:"ignore_eos" yaml:"ignore_eos"`
|
|
RepeatPenalty float64 `json:"repeat_penalty" yaml:"repeat_penalty"`
|
|
Keep int `json:"n_keep" yaml:"n_keep"`
|
|
|
|
FrequencyPenalty float64 `json:"frequency_penalty" yaml:"frequency_penalty"`
|
|
PresencePenalty float64 `json:"presence_penalty" yaml:"presence_penalty"`
|
|
TFZ float64 `json:"tfz" yaml:"tfz"`
|
|
|
|
TypicalP float64 `json:"typical_p" yaml:"typical_p"`
|
|
Seed *int `json:"seed" yaml:"seed"`
|
|
|
|
NegativePrompt string `json:"negative_prompt" yaml:"negative_prompt"`
|
|
RopeFreqBase float32 `json:"rope_freq_base" yaml:"rope_freq_base"`
|
|
RopeFreqScale float32 `json:"rope_freq_scale" yaml:"rope_freq_scale"`
|
|
NegativePromptScale float32 `json:"negative_prompt_scale" yaml:"negative_prompt_scale"`
|
|
// AutoGPTQ
|
|
UseFastTokenizer bool `json:"use_fast_tokenizer" yaml:"use_fast_tokenizer"`
|
|
|
|
// Diffusers
|
|
ClipSkip int `json:"clip_skip" yaml:"clip_skip"`
|
|
|
|
// RWKV (?)
|
|
Tokenizer string `json:"tokenizer" yaml:"tokenizer"`
|
|
}
|