| from huggingface_hub import snapshot_download | |
| from pathlib import Path | |
| import os | |
| API_TKN = os.environ['HF_API_KEY'] | |
| mistral_models_path = 'TanyaVM/7B-Instruct-v0.3' | |
| cache_dir = "/mistral_models/7B-Instruct-v0.3/.cache/huggingface" | |
| os.makedirs(cache_dir, exist_ok=True) | |
| #cache_dir = "\.cache\huggingface" | |
| snapshot_download(repo_id="TanyaVM/mistral_instruct_gen", token = API_TKN,allow_patterns=["params.json", "consolidated.safetensors", "tokenizer.model.v3"],cache_dir=cache_dir) | |