File size: 484 Bytes
a14673a
 
11981d8
 
f309935
11981d8
be4ecac
4124670
2dfdd31
4d2a0d8
4124670
1ea2021
1
2
3
4
5
6
7
8
9
10
11
12
13
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)