dearwolf/LocalAI
1
0
Fork 0
mirror of https://github.com/mudler/LocalAI.git synced 2025-05-24 04:25:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 12

Small fixup to template loading

Browse source
This commit is contained in:
mudler 2023-04-08 11:59:40 +02:00
parent b710147b95
commit 4a932483e1
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
model_loader.go
View file

@ -62,6 +62,7 @@ func (ml *ModelLoader) LoadModel(modelName string, opts ...llama.ModelOption) (*
if _, err := os.Stat(modelBin); os.IsNotExist(err) {
return nil, err
} else {
modelName = fmt.Sprintf("%s.bin", modelName)
modelFile = modelBin
}
}
@ -87,7 +88,7 @@ func (ml *ModelLoader) LoadModel(modelName string, opts ...llama.ModelOption) (*
if err != nil {
return nil, err
}
ml.promptsTemplates[modelFile] = tmpl
ml.promptsTemplates[modelName] = tmpl
}
ml.models[modelFile] = model