Update download.py

Update: download.py

download.py

@@ -1,9 +1,9 @@
-from huggingface_hub import hf_hub_download
-import joblib
+import os
 
-REPO_ID = "KitHung/InternVL_food_lr35_ep10"
+def download_assist_tuner(model_name_or_path):
+    # 设置环境变量
+    os.environ['HF_ENDPOINT'] = 'https://hf-mirror.com'
 
-def download_model(model_path):
-    joblib.load(
-        hf_hub_download(repo_id=REPO_ID, filename=model_path)
-    )
+    # 下载模型
+    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}')