download_models.py 431 Bytes
Newer Older
Rayyyyy's avatar
Rayyyyy committed
1
from modelscope import snapshot_download
Rayyyyy's avatar
Rayyyyy committed
2
3
4
5
# 下载模型可选为:
# LLM-Research/Meta-Llama-3-8B、LLM-Research/Meta-Llama-3-8B-Instruct
# LLM-Research/Meta-Llama-3-70B、LLM-Research/Meta-Llama-3-70B-Instruct

Rayyyyy's avatar
Rayyyyy committed
6
# 下面以 LLM-Research/Meta-Llama-3-8B-Instruct 为例, cache_dir修改为保存模型路径
Rayyyyy's avatar
Rayyyyy committed
7
model_dir = snapshot_download('LLM-Research/Meta-Llama-3-8B-Instruct', cache_dir="/your_model_save_path/")
Rayyyyy's avatar
Rayyyyy committed
8
print(model_dir)