submit 12 models on ppu_zw_810e for zhoushasha
This commit is contained in:
@@ -2,12 +2,12 @@
|
|||||||
|
|
||||||
批量向 ModelHub XC 平台提交模型验证任务的策略服务,之后保持 HTTP 服务存活供平台探活。
|
批量向 ModelHub XC 平台提交模型验证任务的策略服务,之后保持 HTTP 服务存活供平台探活。
|
||||||
|
|
||||||
当前批次使用 2026-09-24 的成功下载模型清单,经非量化过滤后,要求目标 GPU 无验证记录、至少一张其他 GPU 的状态为「已验证」。仅提交 `ppu_zw_810e` 的 17 个模型和 `hygon_k100-ai` 的 321 个模型,共 338 个任务;上一批模型不会重复提交。
|
当前批次仅向 `zhoushasha` 账号提交 `main.py` 中列出的 12 个模型,每个模型只提交 `ppu_zw_810e` 一张卡,共 12 个任务。
|
||||||
|
|
||||||
## 功能
|
## 功能
|
||||||
|
|
||||||
- 使用 `AUTH_TOKEN` 环境变量(未提供时使用 `main.py` 中的预设 Bearer Token)向 `zhoushasha` 账号提交任务
|
- 使用 `AUTH_TOKEN` 环境变量(未提供时使用 `main.py` 中的预设 Bearer Token)向 `zhoushasha` 账号提交任务
|
||||||
- 分 GPU 读取 `model_ids/` 中的模型清单,批量提交 vLLM 验证任务
|
- 从 `main.py` 的 `PPU_MODELS` 读取本批次模型,批量提交验证任务
|
||||||
- 账号额度暂满时,每 30 分钟重试尚未提交的模型
|
- 账号额度暂满时,每 30 分钟重试尚未提交的模型
|
||||||
- 提交结果写入 `submitted_validation_tasks.txt`
|
- 提交结果写入 `submitted_validation_tasks.txt`
|
||||||
- 暴露 `/health` 和 `/status` 接口满足平台运行时契约
|
- 暴露 `/health` 和 `/status` 接口满足平台运行时契约
|
||||||
@@ -17,7 +17,7 @@
|
|||||||
```
|
```
|
||||||
.
|
.
|
||||||
├── main.py # 主入口:HTTP 服务 + 提交逻辑
|
├── main.py # 主入口:HTTP 服务 + 提交逻辑
|
||||||
├── model_ids/ # 本批次两张 GPU 的模型 ID 清单
|
├── model_ids/ # 历史批次模型 ID 清单,本轮未使用
|
||||||
├── Dockerfile # 平台镜像构建配置
|
├── Dockerfile # 平台镜像构建配置
|
||||||
├── requirements.txt # Python 依赖
|
├── requirements.txt # Python 依赖
|
||||||
└── submitted_validation_tasks.txt # 运行后自动生成,记录提交结果
|
└── submitted_validation_tasks.txt # 运行后自动生成,记录提交结果
|
||||||
@@ -32,4 +32,4 @@
|
|||||||
- 通过环境变量 `STRATEGY_ID` 获取策略 ID
|
- 通过环境变量 `STRATEGY_ID` 获取策略 ID
|
||||||
- 正确处理 `SIGTERM` 信号,支持优雅停机
|
- 正确处理 `SIGTERM` 信号,支持优雅停机
|
||||||
|
|
||||||
预设 Bearer Token 于 2026-10-01 22:13(北京时间)到期;届时如需再次运行,应更新 `AUTH_TOKEN`。
|
运行前应确认 `AUTH_TOKEN` 有效;若未设置环境变量,则使用 `main.py` 中的预设 Bearer Token。
|
||||||
|
|||||||
23
main.py
23
main.py
@@ -2,7 +2,7 @@
|
|||||||
xc_validation_strategy — 主入口
|
xc_validation_strategy — 主入口
|
||||||
|
|
||||||
启动后针对 GPU_JOBS 中配置的 GPU 卡分别批量提交各自筛选出的模型验证任务。
|
启动后针对 GPU_JOBS 中配置的 GPU 卡分别批量提交各自筛选出的模型验证任务。
|
||||||
本轮仅提交 ppu_zw_810e 和 hygon_k100-ai 的 2026-09-24 非量化候选清单;
|
本轮仅向 zhoushasha 提交指定的 12 个模型在 ppu_zw_810e 上的验证任务;
|
||||||
其他 GPU 的 config_content 模板仍保留,但未列入 GPU_JOBS。
|
其他 GPU 的 config_content 模板仍保留,但未列入 GPU_JOBS。
|
||||||
(/adminApi/async/task/create-contest-task,
|
(/adminApi/async/task/create-contest-task,
|
||||||
Bearer Token 认证),之后保持 HTTP 服务存活。
|
Bearer Token 认证),之后保持 HTTP 服务存活。
|
||||||
@@ -34,7 +34,7 @@ BASE_URL = os.environ.get("BASE_URL", "https://modelhub.org.cn")
|
|||||||
SUBMIT_ENDPOINT = "/adminApi/async/task/create-contest-task"
|
SUBMIT_ENDPOINT = "/adminApi/async/task/create-contest-task"
|
||||||
|
|
||||||
# 通过 curl -X POST https://modelhub.org.cn/adminApi/user/login 获取后填入
|
# 通过 curl -X POST https://modelhub.org.cn/adminApi/user/login 获取后填入
|
||||||
AUTH_TOKEN = os.environ.get("AUTH_TOKEN", "eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJ1c2VyQWNjb3VudCI6Inpob3VzaGFzaGEiLCJpZCI6MTQsInVzZXJSb2xlIjoibGVhZGVyYm9hcmQiLCJleHAiOjE3OTA4NjQwMDMsImlhdCI6MTc5MDI1OTIwM30.T23Tp3xcI8kkKIOwRCXmZlpe3Qo3sOIxZ8n6NbzJJ2M")
|
AUTH_TOKEN = os.environ.get("AUTH_TOKEN", "eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJ1c2VyQWNjb3VudCI6Inpob3VzaGFzaGEiLCJpZCI6MTQsInVzZXJSb2xlIjoibGVhZGVyYm9hcmQiLCJleHAiOjE3OTIxNDY1MDgsImlhdCI6MTc5MTU0MTcwOH0.bSjmiEg9zcV_wizvak1MUI_XD_PWjXNQ2390JZRqJsU")
|
||||||
CONTEST_API_TOKEN = "ef1ef82f3c9efee413d602345fbe224d"
|
CONTEST_API_TOKEN = "ef1ef82f3c9efee413d602345fbe224d"
|
||||||
CONTRIBUTORS = "zhoushasha"
|
CONTRIBUTORS = "zhoushasha"
|
||||||
TASK_TYPE = "text-generation"
|
TASK_TYPE = "text-generation"
|
||||||
@@ -76,13 +76,24 @@ HYGON_MODELS = load_model_ids("hygon_k100-ai_2026-09-24.txt")
|
|||||||
KUNLUNXIN_MODELS = [
|
KUNLUNXIN_MODELS = [
|
||||||
]
|
]
|
||||||
|
|
||||||
PPU_MODELS = load_model_ids("ppu_zw_810e_2026-09-24.txt")
|
PPU_MODELS = [
|
||||||
|
"raalr/Qwen2.5-1.5-uld-gemma-27b-3",
|
||||||
|
"gemasphi/real-setfit-ss-distiluse-base-multilingual-cased-v2",
|
||||||
|
"intfloat/e5-small",
|
||||||
|
"Nhat1904/32-shot-twitter-2classes-head-body",
|
||||||
|
"Nhat1904/16-shot-twitter-2classes-head-body",
|
||||||
|
"Nhat1904/16-shot-twitter-2classes-new-1",
|
||||||
|
"Nhat1904/16-shot-twitter-2classes-new",
|
||||||
|
"Nhat1904/16-shot-twitter-2classes",
|
||||||
|
"Nhat1904/32-shot-twitter-10epochs",
|
||||||
|
"Nhat1904/32-shot-twitter",
|
||||||
|
"Nhat1904/16-shot-twitter",
|
||||||
|
"kulibinai/cadreasoner",
|
||||||
|
]
|
||||||
|
|
||||||
# 本轮提交:2026-09-24 已成功下载、非量化、目标卡无验证记录,且其他卡至少一张「已验证」的模型。
|
# 本轮仅提交用户指定的 12 个模型,每个模型只提交 ppu_zw_810e。
|
||||||
# 仅 ppu_zw_810e(17) 和 hygon_k100-ai(321),共 338 个;不重复提交上一轮的 24 个任务。
|
|
||||||
GPU_JOBS: List[Tuple[str, List[str]]] = [
|
GPU_JOBS: List[Tuple[str, List[str]]] = [
|
||||||
("ppu_zw_810e", PPU_MODELS),
|
("ppu_zw_810e", PPU_MODELS),
|
||||||
("hygon_k100-ai", HYGON_MODELS),
|
|
||||||
]
|
]
|
||||||
TOTAL_MODELS = sum(len(models) for _, models in GPU_JOBS)
|
TOTAL_MODELS = sum(len(models) for _, models in GPU_JOBS)
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user