fix model name not matching

0625e805 · Michael Yang · c38ec5be · 0625e805
Commit 0625e805 authored Sep 26, 2023 by Michael Yang
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 6 deletions

cmd/cmd.go cmd/cmd.go +2 -6

No files found.
--- a/cmd/cmd.go
+++ b/cmd/cmd.go
@@ -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)
 		}
 	}