From e43b3f8aa36c198b95c1e46bea2eaf3893b13dc3 Mon Sep 17 00:00:00 2001 From: YurenHao0426 Date: Wed, 17 Dec 2025 04:29:37 -0600 Subject: Initial commit (clean history) --- scripts/download_llama.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) create mode 100644 scripts/download_llama.py (limited to 'scripts/download_llama.py') diff --git a/scripts/download_llama.py b/scripts/download_llama.py new file mode 100644 index 0000000..47c7243 --- /dev/null +++ b/scripts/download_llama.py @@ -0,0 +1,16 @@ +from huggingface_hub import snapshot_download +import os + +model_id = "meta-llama/Llama-3.1-8B-Instruct" +local_dir = "models/llama-3.1-8b-instruct" + +os.makedirs(local_dir, exist_ok=True) + +print(f"Downloading {model_id} to {local_dir}...") +snapshot_download( + repo_id=model_id, + local_dir=local_dir, + # token=os.getenv("HF_TOKEN"), # Assuming token is in env or cached +) +print("Download complete.") + -- cgit v1.2.3