Skip to content

GitLab

Commit 536ff4d8 authored by Bruce MacDonald's avatar Bruce MacDonald
Browse files

save file as model name

parent 52beb0a9
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 4 deletions
ollama/model.py
View file @ 536ff4d8
...@@ -49,12 +49,10 @@ def pull(remote, models_home=".", *args, **kwargs): ...@@ -49,12 +49,10 @@ def pull(remote, models_home=".", *args, **kwargs):
if file_info.get("type") == "file" and file_info.get("path").endswith(".bin"): if file_info.get("type") == "file" and file_info.get("path").endswith(".bin"):
f_path = file_info.get("path") f_path = file_info.get("path")
download_url = f"https://huggingface.co/{model}/resolve/{branch}/{f_path}" download_url = f"https://huggingface.co/{model}/resolve/{branch}/{f_path}"
local_filename = os.path.join( local_filename = os.path.join(models_home, os.path.basename(model)) + ".bin"
models_home, os.path.basename(file_info.get("path"))
)
if os.path.exists(local_filename): if os.path.exists(local_filename):
# TODO: check if the file is the same # TODO: check if the file is the same SHA
break break
response = requests.get(download_url, stream=True) response = requests.get(download_url, stream=True)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment