chore: narrow GPU targets, increase crawler pages to 200

app/scheduler/submitter.py

@@ -1,6 +1,9 @@
 from app.clients.modelhub import ModelHubClient
 from app.domain.vllm_configs import gen_vllm_config
 from app.storage.repositories import Repository, future_seconds
+import logging
+
+LOG = logging.getLogger(__name__)
 
 
 class Submitter:
@@ -13,6 +16,7 @@ class Submitter:
         for task in self.repo.due_tasks(("ready_to_submit",), limit=limit):
             config = gen_vllm_config(task["gpu_type"], task["model_id"], task["max_model_len"])
             result = self.client.create_contest_task(task["model_id"], task["gpu_type"], config)
+            LOG.info("submit %s / %s / %s -> %s (%s)", task["model_id"], task["gpu_type"], task["source"], result.result, result.message)
             self.repo.record_submission_attempt(
                 task["id"],
                 task["model_id"],

config.local.json

@@ -4,8 +4,8 @@
   "CONFTEST_API_TOKEN": "0f3c1fd38fbac493cf3760ce4aaeac90",
   "EMAIL": "i-hengyan@4paradigm.com",
   "USER_ID": "83",
-  "CONTRIBUTORS": "i-hengyan",
+  "CONTRIBUTORS": "hengyan",
-  "STRATEGY_ID": "2e5ee92b-ea07-487a-bcfd-02e51be7f6bb",
+  "STRATEGY_ID": "a2514012-5ac8-4187-a3dd-8d74d8582827",
 
   "MODELHUB_URL": "https://modelhub.org.cn",
   "MODELHUB_API_BASE": "https://modelhub.org.cn/api",
@@ -15,7 +15,7 @@
   "HOST": "0.0.0.0",
   "PORT": 8080,
 
-  "SUBMIT_DRY_RUN": true,
+  "SUBMIT_DRY_RUN": false,
   "INJECT_STRATEGY_ID": true,
   "CONTEST_TASK_STRATEGY_FIELD": "strategyId",
 
@@ -25,18 +25,18 @@
   "DOWNLOAD_POLL_INTERVAL_SECONDS": 300,
   "CRAWLER_REFRESH_SECONDS": 3600,
 
-  "DEFAULT_GPU_ALIASES": "910b,k100,p800,166m,bi100,bi150,c500,s4000,mrv100,mlu370-x4,mlu370-x8",
+  "DEFAULT_GPU_ALIASES": "c500,s4000,166m,k100,p800",
   "DEFAULT_MAX_MODEL_LEN": 1024,
 
   "ENABLE_NOT_ADAPTED_CRAWLER": true,
-  "TARGET_MACHINE_NAMES": "MTT S4000,Hygon K100,Kunlunxin P800",
+  "TARGET_MACHINE_NAMES": "MetaX C500,MTT S4000,Biren 166M,Hygon K100,Kunlunxin P800",
   "TARGET_TASK_LEVEL": "文本生成",
   "MODELHUB_PAGE_SIZE": 100,
-  "CRAWLER_MAX_PAGES": 0,
+  "CRAWLER_MAX_PAGES": 200,
 
   "ENABLE_DOWNLOAD_SUCCESS_CRAWLER": true,
   "DOWNLOAD_SUCCESS_PAGE_SIZE": 50,
-  "DOWNLOAD_SUCCESS_MAX_PAGES": 0,
+  "DOWNLOAD_SUCCESS_MAX_PAGES": 20,
 
   "BOUNTY_ALLOW_DIRECT_SUBMIT": true,
   "AUTO_LOAD_SEED_PATH": ""