Compare commits
1 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 4f7c1aa48f |
5
main.py
5
main.py
@@ -14,6 +14,7 @@ from http.server import BaseHTTPRequestHandler, ThreadingHTTPServer
|
||||
MAIN = os.getenv("MAIN_HOST", "https://modelhub.org.cn")
|
||||
GITEA = os.getenv("GITEA_HOST", "https://dev.modelhub.org.cn")
|
||||
TOKEN = os.getenv("EXTERNAL_SERVICE_TOKEN", "") # = xcToken, 走 Xc-Token 头
|
||||
STRATEGY_ID = os.getenv("STRATEGY_ID", "") # 平台注入的自身策略 id, 提交必带 (docs/6 §3)
|
||||
DRY_RUN = os.getenv("DRY_RUN", "false").lower() == "true" # 默认真提交(平台唯一认可路径)
|
||||
TARGET_CARDS = os.getenv("TARGET_CARDS", "Iluvatar_bi-150,Iluvatar_bi-100,Iluvatar_mrv-100").split(",")
|
||||
CARD_NAME = {"Iluvatar_bi-150": "天垓150", "Iluvatar_bi-100": "天垓100", "Iluvatar_mrv-100": "智铠100"}
|
||||
@@ -90,7 +91,7 @@ def submit(model, gpu):
|
||||
if DRY_RUN:
|
||||
return {"model": model.split("/")[-1], "gpu": gpu, "dry_run": True, "cfg_len": len(cfg)}
|
||||
body = json.dumps({"modelAddress": model, "taskType": TASK_TYPE, "targetGpu": gpu,
|
||||
"framework": FRAMEWORK, "configParams": cfg})
|
||||
"framework": FRAMEWORK, "strategyId": STRATEGY_ID, "configParams": cfg})
|
||||
st, txt = _http("POST", MAIN + "/api/adapt/task/add",
|
||||
{"Xc-Token": TOKEN, "Content-Type": "application/json"}, body)
|
||||
try:
|
||||
@@ -103,7 +104,7 @@ def submit(model, gpu):
|
||||
|
||||
|
||||
def loop():
|
||||
log(f"loop start | dry_run={DRY_RUN} | token_present={bool(TOKEN)} | cards={TARGET_CARDS} | type={TASK_TYPE} | max_total={MAX_TOTAL}")
|
||||
log(f"loop start | dry_run={DRY_RUN} | token_present={bool(TOKEN)} | strategy_id={STRATEGY_ID or 'MISSING'} | cards={TARGET_CARDS} | type={TASK_TYPE} | max_total={MAX_TOTAL}")
|
||||
st, j = xc_get("/api/adapt/task/page?current=1&pageSize=1")
|
||||
log(f"auth probe task/page(Xc-Token) -> http={st} code={(j.get('code') if isinstance(j, dict) else '?')}")
|
||||
seen = set()
|
||||
|
||||
Reference in New Issue
Block a user