瀏覽代碼

Update _config_endpoint.py

Xiaomeng Zhao 4 月之前
父節點
當前提交
d3f6736e0a
共有 1 個文件被更改,包括 3 次插入2 次删除
  1. 3 2
      projects/multi_gpu_v2/_config_endpoint.py

+ 3 - 2
projects/multi_gpu_v2/_config_endpoint.py

@@ -15,7 +15,8 @@ def config_endpoint():
     """
 
     os.environ.setdefault('MINERU_MODEL_SOURCE', 'huggingface')
-    model_list_url = f'https://huggingface.co/models'
+    model_list_url = f"https://huggingface.co/models"
+    modelscope_url = f"https://modelscope.cn/models"
     
     # Use a specific check for the Hugging Face source
     if os.environ['MINERU_MODEL_SOURCE'] == 'huggingface':
@@ -56,4 +57,4 @@ def config_endpoint():
     return False
 
 if __name__ == '__main__':
-    print(config_endpoint())
+    print(config_endpoint())