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

Merge pull request #616 from jmorganca/mxyng/fix-model-name

Browse source
This commit is contained in:
Michael Yang 2023-09-26 20:54:18 -07:00 committed by GitHub
commit e6b3648bbf
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
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 return err
} }
modelName, modelTag, ok := strings.Cut(args[0], ":") canonicalModelPath := server.ParseModelPath(args[0])
if !ok {
modelTag = "latest"
}
for _, model := range models.Models { for _, model := range models.Models {
if model.Name == strings.Join([]string{modelName, modelTag}, ":") { if model.Name == canonicalModelPath.GetShortTagname() {
return RunGenerate(cmd, args) return RunGenerate(cmd, args)
} }
} }