foodieproject / download_mode.py
Bunpheng's picture
Create download_mode.py
099d1e6 verified
raw
history blame
505 Bytes
# download_model.py
from huggingface_hub import snapshot_download
from pathlib import Path
def download_model():
mistral_models_path = Path.home().joinpath('mistral_models', '7B-v0.3')
mistral_models_path.mkdir(parents=True, exist_ok=True)
snapshot_download(
repo_id="mistralai/Mistral-7B-v0.3",
allow_patterns=["params.json", "consolidated.safetensors", "tokenizer.model.v3"],
local_dir=mistral_models_path
)
if __name__ == "__main__":
download_model()