From 4537a89b26d74a9e70f2c35137a23369aa86e9d1 Mon Sep 17 00:00:00 2001 From: Bruce MacDonald Date: Wed, 27 Nov 2024 15:22:27 -0800 Subject: [PATCH] Update cmd.go --- cmd/cmd.go | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cmd/cmd.go b/cmd/cmd.go index 90fd4e09f..25f26aab2 100644 --- a/cmd/cmd.go +++ b/cmd/cmd.go @@ -629,7 +629,7 @@ func PushHandler(cmd *cobra.Command, args []string) error { spinner.Stop() destination := n.String() - if strings.HasSuffix(n.Host, ".ollama.ai") || strings.HasSuffix(n.Host, ".ollama.com") { + if isOllamaHost { destination = "https://ollama.com/" + strings.TrimSuffix(n.DisplayShortest(), ":latest") } fmt.Printf("\nYou can find your model at:\n\n")