feat: expose mirostat to config (#193)

This commit is contained in:
Ettore Di Giacinto 2023-05-05 13:45:37 +02:00 committed by GitHub
parent c839b334eb
commit 961cf29217
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 31 additions and 0 deletions

View file

@ -24,6 +24,9 @@ type Config struct {
Embeddings bool `yaml:"embeddings"`
Backend string `yaml:"backend"`
TemplateConfig TemplateConfig `yaml:"template"`
MirostatETA float64 `yaml:"mirostat_eta"`
MirostatTAU float64 `yaml:"mirostat_tau"`
Mirostat int `yaml:"mirostat"`
}
type TemplateConfig struct {