highperfocused/ollama
1
0
Fork 0
mirror of https://github.com/ollama/ollama.git synced 2025-08-27 19:23:14 +02:00
Code Issues Packages Projects Releases 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
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"):
f_path = file_info.get("path")
download_url = f"https://huggingface.co/{model}/resolve/{branch}/{f_path}"
local_filename = os.path.join(
models_home, os.path.basename(file_info.get("path"))
)
local_filename = os.path.join(models_home, os.path.basename(model)) + ".bin"
if os.path.exists(local_filename):
# TODO: check if the file is the same
# TODO: check if the file is the same SHA
break
response = requests.get(download_url, stream=True)