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

save file as model name

parent 52beb0a9
......@@ -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)
......
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