Views
No views yet
1from huggingface_hub import snapshot_download
2from huggingface_hub import login
3import dotenv
4import os
5dotenv.load_dotenv()
6login(token=os.getenv("HF_TOKEN"))
7
8repo_id = "deepseek-ai/DeepSeek-R1-0528"
9local_dir = "/scratch/vladimir_albrekht/projects/smollm/synthetic_data_generation/models/DeepSeek-R1-0528"
10
11
12def download_model_repo(repo_id, local_dir):
13 print(f"Downloading repository {repo_id}...")
14
15 # Download the complete repository
16 snapshot_download(
17 repo_id=repo_id,
18 local_dir=local_dir
19 # allow_patterns=["DeepSeek-R1-Q6_K/*"]
20 )
21 print(f"Repository downloaded to {local_dir}")
22
23# Specify the model repo ID, save directory, and Huggingw Face token
24
25# Download the repository
26download_model_repo(repo_id, local_dir)