Mirror/LocalAI
1
0
Fork 0
mirror of https://github.com/mudler/LocalAI.git synced 2024-06-07 19:40:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

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
Show Stats 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