matt/ollama37
1
0
Fork 0
mirror of https://github.com/dogkeeper886/ollama37.git synced 2025-12-12 08:47:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix pull model name

Browse Source
This commit is contained in:
Michael Yang
2023-06-30 11:54:46 -07:00
parent e69a08f1b1
commit 3ce3caaf65
2 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
ollama/cmd/cli.py
View File

@@ -151,7 +151,7 @@ def pull(*args, **kwargs):
def run(*args, **kwargs):
name = model.pull(*args, **kwargs)
name = model.pull(model_name=kwargs.pop('model'), *args, **kwargs)
kwargs.update({"model": name})
print(f"Running {name}...")
generate(*args, **kwargs)