baalajimaestro/ollama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Merge pull request #420 from jmorganca/mxyng/34b-mem-check

Browse source 
add 34b to mem check
This commit is contained in:
Michael Yang 2023-08-26 14:15:52 -07:00 committed by GitHub
commit e82fcf30c6
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
llm/llm.go
View file

@ -55,7 +55,7 @@ func New(model string, adapters []string, opts api.Options) (LLM, error) {
if totalResidentMemory < 16*1024*1024 { if totalResidentMemory < 16*1024*1024 {
return nil, fmt.Errorf("model requires at least 16GB of memory") return nil, fmt.Errorf("model requires at least 16GB of memory")
} }
case ModelType30B: case ModelType30B, ModelType34B:
if totalResidentMemory < 32*1024*1024 { if totalResidentMemory < 32*1024*1024 {
return nil, fmt.Errorf("model requires at least 32GB of memory") return nil, fmt.Errorf("model requires at least 32GB of memory")
} }