InternVL / download.py
Kit-Hung
Update: download
580d757
raw
history blame contribute delete
320 Bytes
import os
def download_model(model_name_or_path):
# 设置环境变量
os.environ['HF_ENDPOINT'] = 'https://hf-mirror.com'
# 下载模型
os.system(
f'mkdir -p {model_name_or_path} && huggingface-cli download --resume-download KitHung/InternVL_food_lr35_ep10 --local-dir {model_name_or_path}')