Downloads · 30 days
1.2K
76% of all-time downloads
wam_ctxpool_avg
hmkang/wam_ctxpool_avg
wam_ctxpool_avg is a robotics model from hmkang. Use it for the robotics task on the model card, and read the license before you ship it in a product. It is set up for wan2.2. The card lists the license as apache-2.0.
Wan2.2-TI2V-5B video DiT fine-tuned on RoboCasa (base recipe, effective batch 64: 4 GPU × per-device 8 × grad-accum 2), 4-latin history (nin=25 / nout=41, fdf=2), with average context pooling of the past latent frames…
All-time downloads
1.6K
Public
Repo size
628 GB
Likes
1
Public
Repo makeup
Click a slice to open those files.
.safetensors628 GB · 100%
At a glance
- Task
- Robotics
- Library
- wan2.2
- License
- apache-2.0
- Access
- Public
- Created
- Sep 5, 2026
- Updated
- Sep 12, 2026
- SHA
- a6803506
Model card
From the Hugging Face model README
WAM_DIT4DIT — context pooling (avg), RoboCasa kitchen, video-only
Wan2.2-TI2V-5B video DiT fine-tuned on RoboCasa (base recipe, effective batch 64:
4 GPU × per-device 8 × grad-accum 2), 4-latin history (nin=25 / nout=41, fdf=2),
with average context pooling of the past latent frames inserted before block L.
| folder | pooling | layer | steps |
|---|---|---|---|
L12/checkpoint-<step> | avg | 12 | every 20k |
L3/checkpoint-<step> | avg | 3 | every 20k |
Weights (*.safetensors) + config.json + processor / experiment config only; optimizer
state and training_args.bin are not included. Training code: https://github.com/HEMMO0208/wam
(run_scripts/train/wam_dit4dit/compression/finetune_wam_dit4dit_robocasa_kitchen_ctxpool.sh).
Note: an earlier version of this repo held effective-batch-32 runs; those were removed. All checkpoints here are eff-64.
- Task
- Robotics
- Library
- wan2.2
- License
- apache-2.0
- Created
- Sep 5, 2026
- Updated
- Sep 12, 2026