# 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()