ExternalVendorCode/LocalAI
1
0
Fork 0
mirror of https://github.com/mudler/LocalAI.git synced 2025-05-07 11:08:27 +00:00
Code Issues Actions 12 Packages Projects Releases Wiki Activity

fix(llama.cpp): set -1 as default for max tokens (#2087)

Browse Source 
Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
This commit is contained in:
Ettore Di Giacinto 2024-04-20 20:20:10 +02:00 committed by GitHub
parent 03adc1f60d
commit afa1bca1e3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
core/config/backend_config.go
View File

@ -205,15 +205,15 @@ func (cfg *BackendConfig) SetDefaults(opts ...ConfigLoaderOption) {
defaultTopP := 0.95 defaultTopP := 0.95
defaultTopK := 40 defaultTopK := 40
defaultTemp := 0.9 defaultTemp := 0.9
defaultMaxTokens := 2048
defaultMirostat := 2 defaultMirostat := 2
defaultMirostatTAU := 5.0 defaultMirostatTAU := 5.0
defaultMirostatETA := 0.1 defaultMirostatETA := 0.1
defaultTypicalP := 1.0 defaultTypicalP := 1.0
defaultTFZ := 1.0 defaultTFZ := 1.0
defaultInfinity := -1
// Try to offload all GPU layers (if GPU is found) // Try to offload all GPU layers (if GPU is found)
defaultNGPULayers := 99999999 defaultHigh := 99999999
trueV := true trueV := true
falseV := false falseV := false
@ -254,7 +254,7 @@ func (cfg *BackendConfig) SetDefaults(opts ...ConfigLoaderOption) {
} }
if cfg.Maxtokens == nil { if cfg.Maxtokens == nil {
cfg.Maxtokens = &defaultMaxTokens cfg.Maxtokens = &defaultInfinity
} }
if cfg.Mirostat == nil { if cfg.Mirostat == nil {
@ -269,7 +269,7 @@ func (cfg *BackendConfig) SetDefaults(opts ...ConfigLoaderOption) {
cfg.MirostatTAU = &defaultMirostatTAU cfg.MirostatTAU = &defaultMirostatTAU
} }
if cfg.NGPULayers == nil { if cfg.NGPULayers == nil {
cfg.NGPULayers = &defaultNGPULayers cfg.NGPULayers = &defaultHigh
} }
if cfg.LowVRAM == nil { if cfg.LowVRAM == nil {