Skip to content

Commit

Permalink
Merge pull request #416 from bigstomach/patch-1
Browse files Browse the repository at this point in the history
fix: model tag issue in Ollama
  • Loading branch information
zainhoda authored May 6, 2024
2 parents 9e22c72 + c641ae5 commit ef7eb72
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/vanna/ollama/ollama.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@ def __init__(self, config=None):
raise ValueError("config must contain at least Ollama model")
self.host = config.get("ollama_host", "http://localhost:11434")
self.model = config["model"]
if ":" in self.model:
if ":" not in self.model:
self.model += ":latest"

self.ollama_client = ollama.Client(self.host, timeout=Timeout(240.0))
Expand Down

0 comments on commit ef7eb72

Please sign in to comment.