baalajimaestro/ollama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

save file as model name

Browse source
This commit is contained in:
Bruce MacDonald 2023-06-28 14:26:19 -04:00
parent 52beb0a99e
commit 536ff4d816
1 changed files with 2 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
ollama/model.py
View file

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