OpenMOSS/EasyWAM-MoT-Wan22-LoRA-128
模型介绍
文件和版本
Pull Requests
讨论
分析

EasyWAM-MoT-Wan22-LoRA-128

EasyWAM-MoT 采用相互独立的 Video DiT 与 Action DiT 专家,二者的 token 经由混合自注意力实现交互。该检查点在 LIBERO 上使用秩 128 的 LoRA(r=128、alpha=128)训练,并以 Wan2.2-TI2V-5B 作为视频骨干网络。

相应检查点基于 EasyWAM 代码库训练获得。

结果

在 EasyWAM LIBERO 评估协议下的成功率(%):

模型空间物体目标长程平均
全参数
EasyWAM-Unified99.099.499.298.299.0
EasyWAM-MoT97.898.497.695.697.4
EasyWAM-Hidden99.4100.097.097.898.6
LoRA(秩 128)
EasyWAM-Unified84.097.892.081.288.8
🔥 EasyWAM-MoT96.898.894.490.495.1
EasyWAM-Hidden96.899.492.686.893.9

在 LIBERO-Plus 评估协议下的成功率(%):

模型背景相机语言布局光照噪声机器人平均
EasyWAM-Unified55.833.793.780.692.250.271.467.5
EasyWAM-MoT52.820.680.465.285.151.549.756.8
EasyWAM-Hidden56.849.295.381.090.458.277.472.4

下载

hf download OpenMOSS-Team/EasyWAM-MoT-Wan22-LoRA-128 \
  easywam_mot_wan22_lora_128.pt \
  --local-dir ./checkpoints

评估

按照 EasyWAM LIBERO 指南 中的说明,准备 Wan2.2、LIBERO 以及对应的 dataset_stats.json,然后运行:

python experiments/libero/run_libero_manager.py \
  task=libero_easywam_mot_wan22_lora \
  ckpt=./checkpoints/easywam_mot_wan22_lora_128.pt \
  EVALUATION.dataset_stats_path=<path-to-matching-dataset_stats.json>

EasyWAM 会根据任务配置创建对应的 LoRA 模块,并在加载该检查点时将其合并用于评估。

检查点详情

  • 架构:EasyWAM-MoT
  • 主干:Wan2.2-TI2V-5B
  • 训练:LoRA,rank 128,alpha 128
  • 数据集:LIBERO,双相机,224 px
  • 训练步数:20,000
  • 动作维度:7
  • 状态维度:8
  • 格式:EasyWAM PyTorch checkpoint(.pt)

许可证与引用

EasyWAM 代码以 MIT License 发布。使用此检查点还需遵守其基础模型和训练数据的相关条款。如果 EasyWAM 对您的研究有帮助,请引用:

@misc{easywam2026,
  title  = {EasyWAM: A Unified and Efficient Framework for Training and Evaluating World Action Models},
  author = {EasyWAM-Team},
  year   = {2026},
  url    = {https://github.com/OpenMOSS/EasyWAM}
}