JBAIP / download_model.py
Diamanta's picture
Rename download_mode.py to download_model.py
00d1b14 verified
raw
history blame
279 Bytes
from huggingface_hub import hf_hub_download
import os
repo_id = os.environ.get("MODEL_REPO")
filename = os.environ.get("MODEL_FILE")
print("Downloading model:", repo_id, filename)
hf_hub_download(repo_id=repo_id, filename=filename, local_dir=".", local_dir_use_symlinks=False)