Update cmd.go

This commit is contained in:
Bruce MacDonald 2024-11-27 15:22:27 -08:00
parent 85822544a9
commit 4537a89b26

View File

@ -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")