Xiaomeng Zhao 4 месяцев назад
Родитель
Сommit
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())