Robotics
PyTorch
world-action-model
libero
Viani commited on
Commit
d9ddc71
·
verified ·
1 Parent(s): be3840a

Restore the statement that the Wan2.2 weights are fine-tuned.

Browse files
Files changed (3) hide show
  1. ASSET_SHA256SUMS +2 -2
  2. NOTICE +4 -6
  3. README.md +4 -4
ASSET_SHA256SUMS CHANGED
@@ -1,8 +1,8 @@
1
  c485bd2ee62dbb1828eea5c5ea2bb7b0178f60d809329ebfd15bdf731f592f9a action_vae.pt
2
  cfc7749b96f63bd31c3c42b5c471bf756814053e847c10f3eb003417bc523d30 LICENSE
3
- 93235892c29664be5451bec9c282ccf661bba6f6fc03a017bf6a00ec9d1a06d2 NOTICE
4
  b2cd77e9402b7897ba0d680eb45e4a38fb752e89641e3a9487a7cd4e947dad6c policy.pt
5
- b460699373f8223f35d80b5da30863f225da0557bae4a92e9bfcf4ea46f44b56 README.md
6
  835a23bad8ec61cf2c27c2f9440fe1f902ae5365c29f0c1858c0cb646e883594 stats/libero_10.json
7
  a648e545ad63d0d398ebf5735aa0e7426efdaf3fa55907c5ec50b72df6e273df stats/libero_goal.json
8
  4eed60c361514c0f95f4ad24ec271972197e30463be07d82ec6c764cba3f6f83 stats/libero_object.json
 
1
  c485bd2ee62dbb1828eea5c5ea2bb7b0178f60d809329ebfd15bdf731f592f9a action_vae.pt
2
  cfc7749b96f63bd31c3c42b5c471bf756814053e847c10f3eb003417bc523d30 LICENSE
3
+ b19048e39168d2101fa3330edc9574e09dc86b75bd3b7f6046f91b95ba508694 NOTICE
4
  b2cd77e9402b7897ba0d680eb45e4a38fb752e89641e3a9487a7cd4e947dad6c policy.pt
5
+ b4e53b1106e288caa42fd71d9ad8e8e8cf80c19b947442aef569be274dad6fe0 README.md
6
  835a23bad8ec61cf2c27c2f9440fe1f902ae5365c29f0c1858c0cb646e883594 stats/libero_10.json
7
  a648e545ad63d0d398ebf5735aa0e7426efdaf3fa55907c5ec50b72df6e273df stats/libero_goal.json
8
  4eed60c361514c0f95f4ad24ec271972197e30463be07d82ec6c764cba3f6f83 stats/libero_object.json
NOTICE CHANGED
@@ -1,13 +1,11 @@
1
  FastWAM LIBERO checkpoints
2
  Copyright (c) 2026 Advanced Micro Devices, Inc.
3
 
4
- These checkpoints include Wan2.2-TI2V-5B
5
  (https://huggingface.co/Wan-AI/Wan2.2-TI2V-5B, Wan Team, Alibaba), licensed
6
- under the Apache License, Version 2.0. The Wan2.2-TI2V-5B weights are the
7
- published weights and are unchanged. No data was used to train them
8
- separately. The action policy was trained with the FastWAM recipe
9
  (https://github.com/yuantianyuan01/FastWAM, MIT License).
10
 
11
  Training data: LIBERO (Liu et al., NeurIPS 2023; CC BY 4.0), in the LeRobot
12
- format published as yuanty/LIBERO-fastwam (CC BY 4.0). The dataset is used as
13
- published. No changes were made to the training data.
 
1
  FastWAM LIBERO checkpoints
2
  Copyright (c) 2026 Advanced Micro Devices, Inc.
3
 
4
+ These checkpoints contain a modified version of Wan2.2-TI2V-5B
5
  (https://huggingface.co/Wan-AI/Wan2.2-TI2V-5B, Wan Team, Alibaba), licensed
6
+ under the Apache License, Version 2.0. The Wan2.2-TI2V-5B weights were
7
+ fine-tuned by Advanced Micro Devices, Inc. using the FastWAM training recipe
 
8
  (https://github.com/yuantianyuan01/FastWAM, MIT License).
9
 
10
  Training data: LIBERO (Liu et al., NeurIPS 2023; CC BY 4.0), in the LeRobot
11
+ format published as yuanty/LIBERO-fastwam (CC BY 4.0).
 
README.md CHANGED
@@ -12,7 +12,7 @@ tags:
12
 
13
  # FastWAM LIBERO
14
 
15
- A latent world-action policy for the [LIBERO](https://github.com/Lifelong-Robot-Learning/LIBERO) benchmark from Advanced Micro Devices, Inc. It uses the published [Wan2.2-TI2V-5B](https://huggingface.co/Wan-AI/Wan2.2-TI2V-5B) weights unchanged.
16
 
17
  Each 32-step action chunk is encoded as 8 latent tokens of width 16. On the four LIBERO suites the policy reaches **97.2%** average success with a single sample.
18
 
@@ -44,7 +44,7 @@ huggingface-cli download Wan-AI/Wan2.1-T2V-1.3B \
44
  --local-dir checkpoints/Wan-AI/Wan2.1-T2V-1.3B
45
  ```
46
 
47
- The full Wan video DiT is not required. `policy.pt` already contains the published Wan2.2 video weights and the trained action expert.
48
 
49
  LIBERO simulator assets, BDDL task files, and initial states are also required.
50
 
@@ -62,9 +62,9 @@ Research on world-action models and robot-policy inference, in simulation.
62
 
63
  These weights are licensed under the Apache License 2.0 ([`LICENSE`](LICENSE)). See [`NOTICE`](NOTICE).
64
 
65
- The [Wan2.2-TI2V-5B](https://huggingface.co/Wan-AI/Wan2.2-TI2V-5B) weights included here are the published weights (Apache-2.0, Wan Team, Alibaba). They are unchanged, and no data was used to train them separately. The action policy was trained with the [FastWAM](https://github.com/yuantianyuan01/FastWAM) recipe (MIT). That license covers the training code, not these weights.
66
 
67
- Training data: [LIBERO](https://github.com/Lifelong-Robot-Learning/LIBERO) (CC BY 4.0), in the LeRobot format published as [yuanty/LIBERO-fastwam](https://huggingface.co/datasets/yuanty/LIBERO-fastwam) (CC BY 4.0). This is the dataset used for FastWAM LIBERO, and it is used as published. No changes were made to the training data. The Wan2.2 weights are used unchanged. No data was used to train them separately.
68
 
69
  ## Citation
70
 
 
12
 
13
  # FastWAM LIBERO
14
 
15
+ A latent world-action policy for the [LIBERO](https://github.com/Lifelong-Robot-Learning/LIBERO) benchmark, fine-tuned by Advanced Micro Devices, Inc. from [Wan2.2-TI2V-5B](https://huggingface.co/Wan-AI/Wan2.2-TI2V-5B).
16
 
17
  Each 32-step action chunk is encoded as 8 latent tokens of width 16. On the four LIBERO suites the policy reaches **97.2%** average success with a single sample.
18
 
 
44
  --local-dir checkpoints/Wan-AI/Wan2.1-T2V-1.3B
45
  ```
46
 
47
+ The full Wan video DiT is not required. `policy.pt` already contains the fine-tuned video and action experts.
48
 
49
  LIBERO simulator assets, BDDL task files, and initial states are also required.
50
 
 
62
 
63
  These weights are licensed under the Apache License 2.0 ([`LICENSE`](LICENSE)). See [`NOTICE`](NOTICE).
64
 
65
+ They are a modified version of [Wan2.2-TI2V-5B](https://huggingface.co/Wan-AI/Wan2.2-TI2V-5B) (Apache-2.0, Wan Team, Alibaba), fine-tuned by Advanced Micro Devices, Inc. The training recipe is [FastWAM](https://github.com/yuantianyuan01/FastWAM) (MIT). That license covers the training code, not these weights.
66
 
67
+ Training data: [LIBERO](https://github.com/Lifelong-Robot-Learning/LIBERO) (CC BY 4.0), in the LeRobot format published as [yuanty/LIBERO-fastwam](https://huggingface.co/datasets/yuanty/LIBERO-fastwam) (CC BY 4.0). This is the dataset used for FastWAM LIBERO, and it is used as published. No changes were made to the training data.
68
 
69
  ## Citation
70