highperfocused/ollama
1
0
Fork 0
mirror of https://github.com/ollama/ollama.git synced 2025-04-06 10:58:36 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix model name not matching

Browse Source
This commit is contained in:
Michael Yang 2023-09-26 19:49:55 -07:00
parent c38ec5befb
commit 0625e805f0
1 changed files with 2 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
cmd/cmd.go
View File

@ -127,13 +127,9 @@ func RunHandler(cmd *cobra.Command, args []string) error {
return err
}
modelName, modelTag, ok := strings.Cut(args[0], ":")
if !ok {
modelTag = "latest"
}
canonicalModelPath := server.ParseModelPath(args[0])
for _, model := range models.Models {
if model.Name == strings.Join([]string{modelName, modelTag}, ":") {
if model.Name == canonicalModelPath.GetShortTagname() {
return RunGenerate(cmd, args)
}
}