SolarWM-Wan2.2-5B 三阶段摸底清单

• 来源：聊天摸底清单 2026-09-22
• 状态：进行中

按顺序勾；官方：Stage0.5 FM → Stage1 TF-AnyFlow → Stage2 DMD/SGF

0. 环境与资产（先做完再训）

• [ ] clone Junchao-cs/SolarWM，读 docs/backends/wan22-ti2v-5b.md + environments
• [ ] HF 申请 gated：SolarWM-Wan2.2-5B（或总仓 junchaoh-cs/SolarWM 里 SolarWM-5B-*）
• [ ] 下载并保持目录名不变：
  • SolarWM-5B-base
  • SolarWM-5B-bid-stage0p5-81f（+ 可选 …-153f）
  • SolarWM-5B-tf-stage1-81f
  • SolarWM-5B-sgf-stage2-81f
• [ ] 设好 SOLAR_REPO / SOLAR_MODEL_ROOT / SOLAR_DATA_ROOT / SOLAR_OUTPUT_ROOT
• [ ] 数据：摸底推理可先跟官方 test index；要复训再下 raw-WDS / 153f latent（ModelScope latent 包）
• [ ] license：代码 Apache；权重/基座单独核对 Wan + Solar 说明

1. 推理摸底（本周必交，单卡优先 Stage2）

阶段	权重目录	官方 infer config	GPU（文档）	你要记的
0.5 Bid+相机	…-bid-stage0p5-81f（或 153f）	infer_stage0p5_fm_81f.yaml	8	相机可控、画质、非实时
1 TF-AnyFlow	…-tf-stage1-81f	infer_stage1_tf_anyflow_v1_5_81f.yaml	8	因果化中间态观感
2 SGF 学生	…-sgf-stage2-81f	infer_stage2_sgf_camera_length.yaml	1	Demo 同宗入口：延迟/帧率/相机轨迹接口

Stage2 示例（文档原文形态）：

torchrun --standalone --nproc-per-node=1 -m solarwm infer \
  --config configs/examples/wan22_ti2v_5b/infer_stage2_sgf_camera_length.yaml \
  --set model.base_path=$SOLAR_MODEL_ROOT/SolarWM-5B-base \
  --set checkpoint.path=$SOLAR_MODEL_ROOT/SolarWM-5B-sgf-stage2-81f \
  --set data.index_root=$SOLAR_DATA_ROOT \
  --set data.transport.root=$SOLAR_DATA_ROOT \
  --set inference.run_id=smoke-stage2 \
  --set runtime.output_dir=$SOLAR_OUTPUT_ROOT/wan5-stage2-sgf-infer

每阶段勾完要留下：成功命令、样例 mp4、大致显存/耗时、相机输入长什么样（给 Demo 槽位对齐）。

2. 训练契约摸底（能 resolve config 就算过，不必两周训满）

• [ ] solarwm config resolve 跑通各 stage 的 train yaml
• [ ] 记清初始化链（官方）：
  • Stage1 ← bid-stage0p5-153f（EMA）
  • Stage2 student ← tf-stage1-81f；teacher/critic ← bid-stage0p5-81f
• [ ] 分清：model.pt = 权重初始化；完整 checkpoint_model_* 才可 full-resume（含 RNG/reader；Stage2 还要 critic.pt）
• [ ] 写一页「创新默认插在 Stage1→2」：改什么、不动 serving

3. 和 Demo / 双轨的交接物

• [ ] 模型槽约定：v1 = 官方 sgf-stage2-81f；v1.x = 自蒸同接口学生
• [ ] 列清：官方有相机轨迹 infer，没有 Zing 式 WebSocket 键盘+中途文本——serving 胶水归 Demo
• [ ] H3 只记 TODO：SolarWM-H3-33B 目前公开到 Stage0.5，不挡本周 Wan5B 摸底

4. 本周最小完成线（建议）

1. HF 通过 + 权重落盘
2. Stage2 单卡 infer 出片
3. Stage0.5 / Stage1 至少各跑通一条（卡不够可先读 config + 文档，标阻塞）
4. 半页「阶段 → checkpoint → 给领导的版本号」

相关：SolarWM 卡片 · 训练阶段 · Wan5B 后训对照。