JBAIP / download_mode.py
Diamanta's picture
Create download_mode.py
0f2ad2d 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)