Przeglądaj źródła

Update projects/multi_gpu_v2/_config_endpoint.py

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
Xiaomeng Zhao 4 miesięcy temu
rodzic
commit
07b4cbc0ec
1 zmienionych plików z 1 dodań i 1 usunięć
  1. 1 1
      projects/multi_gpu_v2/_config_endpoint.py

+ 1 - 1
projects/multi_gpu_v2/_config_endpoint.py

@@ -38,7 +38,7 @@ def config_endpoint():
     
     elif os.environ['MINERU_MODEL_SOURCE'] == 'modelscope':
         try:
-            response = requests.head(model_list_url, timeout=TIMEOUT)
+            response = requests.head(modelscope_url, timeout=TIMEOUT)
             if response.ok:
                 logging.info(f"Successfully connected to ModelScope. Using 'modelscope' as model source.")
                 return True