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

resume download of models from directory

Browse Source
This commit is contained in:
Bruce MacDonald
2023-07-04 11:16:50 -04:00
parent 35f202f573
commit a86a4f8c26
2 changed files with 28 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
ollama/cmd/cli.py
View File

@@ -175,12 +175,18 @@ def search(*args, **kwargs):
def pull(*args, **kwargs):
model.pull(model_name=kwargs.pop('model'), *args, **kwargs)
print("Up to date.")
try:
model.pull(model_name=kwargs.pop('model'), *args, **kwargs)
print("Up to date.")
except Exception as e:
print(f"An error occurred: {e}")
def run(*args, **kwargs):
name = model.pull(model_name=kwargs.pop('model'), *args, **kwargs)
kwargs.update({"model": name})
print(f"Running {name}...")
generate(*args, **kwargs)
try:
name = model.pull(model_name=kwargs.pop('model'), *args, **kwargs)
kwargs.update({"model": name})
print(f"Running {name}...")
generate(*args, **kwargs)
except Exception as e:
print(f"An error occurred: {e}")