1 files changed, 16 insertions, 0 deletions
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.")
+