JBAIP / download_model.py
Diamanta's picture
Update download_model.py
c81cd55 verified
raw
history blame contribute delete
506 Bytes
import os
from huggingface_hub import hf_hub_download
# Set environment to use writable location
os.environ["HF_HOME"] = "/tmp/huggingface"
os.environ["HF_HUB_CACHE"] = "/tmp/huggingface"
# Download model into temp-writable location
model_path = hf_hub_download(
repo_id="TheBloke/CodeLlama-7B-Instruct-GGUF",
filename="codellama-7b-instruct.Q4_K_M.gguf",
cache_dir="/tmp/huggingface"
)
# Save the path so app.py can use it
with open("/tmp/model_path.txt", "w") as f:
f.write(model_path)