docs: add RIFT model card
Browse files
README.md
ADDED
|
@@ -0,0 +1,119 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: apache-2.0
|
| 3 |
+
library_name: fastwam
|
| 4 |
+
pipeline_tag: robotics
|
| 5 |
+
base_model: Wan-AI/Wan2.2-TI2V-5B
|
| 6 |
+
base_model_relation: finetune
|
| 7 |
+
datasets:
|
| 8 |
+
- yuanty/LIBERO-fastwam
|
| 9 |
+
tags:
|
| 10 |
+
- robotics
|
| 11 |
+
- world-action-model
|
| 12 |
+
- flow-matching
|
| 13 |
+
- libero
|
| 14 |
+
- pytorch
|
| 15 |
+
---
|
| 16 |
+
|
| 17 |
+
# RIFT
|
| 18 |
+
|
| 19 |
+
This private model repository contains the canonical RIFT checkpoint trained
|
| 20 |
+
on LIBERO. RIFT is derived from
|
| 21 |
+
[FastWAM](https://github.com/yuantianyuan01/FastWAM) at commit
|
| 22 |
+
`45d8e1458921d83f8ad6cf9ce993d371208dabd0`.
|
| 23 |
+
|
| 24 |
+
RIFT uses one representation-shaping flow-matching head. The checkpoint also
|
| 25 |
+
retains a detached L2 diagnostic probe for schema compatibility; this probe
|
| 26 |
+
does not shape the shared anticipation representation.
|
| 27 |
+
|
| 28 |
+
## Files
|
| 29 |
+
|
| 30 |
+
| File | Size | SHA-256 |
|
| 31 |
+
|---|---:|---|
|
| 32 |
+
| `rift_step021700.pt` | 12,060,942,777 bytes | `080e260b26c580cff5da0ce1de4c72e57eb2a6de47ae12f5712244dcf911f3cb` |
|
| 33 |
+
| `dataset_stats.json` | 40,961 bytes | `f68a07ec322ea425164d94a8827aec1430109c7614637d03aec894aadc9fc05b` |
|
| 34 |
+
|
| 35 |
+
`dataset_stats.json` is the training-run normalization file paired with this
|
| 36 |
+
checkpoint. Keep both files in the same directory for automatic discovery.
|
| 37 |
+
|
| 38 |
+
## Source and configuration
|
| 39 |
+
|
| 40 |
+
- Source commit: [`184a582d7821e3a919dcd6e5795110ac0282eda6`](https://github.com/ChushanZhang/RIFT/tree/184a582d7821e3a919dcd6e5795110ac0282eda6)
|
| 41 |
+
- Model configuration: [`config.yaml`](./config.yaml)
|
| 42 |
+
- Canonical training task: [`configs/task/libero_rift_2cam224_1e-4.yaml`](https://github.com/ChushanZhang/RIFT/blob/184a582d7821e3a919dcd6e5795110ac0282eda6/configs/task/libero_rift_2cam224_1e-4.yaml)
|
| 43 |
+
- Training step: 21,700
|
| 44 |
+
- Checkpoint dtype: bfloat16
|
| 45 |
+
|
| 46 |
+
The model depends on `Wan-AI/Wan2.2-TI2V-5B`, the tokenizer/text-encoder
|
| 47 |
+
assets from `Wan-AI/Wan2.1-T2V-1.3B`, and the ActionDiT backbone generated by
|
| 48 |
+
the preprocessing script in the source repository. The original training run
|
| 49 |
+
did not record immutable Hub revisions for these external assets. Current
|
| 50 |
+
dependency revisions must not be treated as proven training-time revisions.
|
| 51 |
+
|
| 52 |
+
## Download
|
| 53 |
+
|
| 54 |
+
```bash
|
| 55 |
+
hf download PoopBear/RIFT \
|
| 56 |
+
rift_step021700.pt dataset_stats.json config.yaml \
|
| 57 |
+
--local-dir checkpoints/rift
|
| 58 |
+
```
|
| 59 |
+
|
| 60 |
+
Because this repository is private, authenticate with `hf auth login` first.
|
| 61 |
+
|
| 62 |
+
Verify the artifacts before loading:
|
| 63 |
+
|
| 64 |
+
```bash
|
| 65 |
+
sha256sum checkpoints/rift/rift_step021700.pt \
|
| 66 |
+
checkpoints/rift/dataset_stats.json
|
| 67 |
+
```
|
| 68 |
+
|
| 69 |
+
## Install the matching source
|
| 70 |
+
|
| 71 |
+
```bash
|
| 72 |
+
git clone https://github.com/ChushanZhang/RIFT.git
|
| 73 |
+
cd RIFT
|
| 74 |
+
git checkout 184a582d7821e3a919dcd6e5795110ac0282eda6
|
| 75 |
+
pip install -e .
|
| 76 |
+
```
|
| 77 |
+
|
| 78 |
+
The weight file contains `mot`, `proprio_encoder`, `step`, and `torch_dtype`.
|
| 79 |
+
It is a weights-only checkpoint: optimizer, scheduler, and dataloader state are
|
| 80 |
+
not included, so it is suitable for evaluation or warm-starting rather than an
|
| 81 |
+
exact training resume.
|
| 82 |
+
|
| 83 |
+
## Inputs and outputs
|
| 84 |
+
|
| 85 |
+
The canonical LIBERO recipe uses two RGB cameras at 224 x 224, 8-dimensional
|
| 86 |
+
proprioception, 7-dimensional actions, and a 32-step action horizon. Outputs
|
| 87 |
+
must be interpreted using the included dataset statistics.
|
| 88 |
+
|
| 89 |
+
## Evaluation status
|
| 90 |
+
|
| 91 |
+
This artifact release does not claim a new benchmark result. Before reporting
|
| 92 |
+
results, pin the simulator commit, task suite, seeds, episodes per task,
|
| 93 |
+
action-chunking policy, and aggregation method.
|
| 94 |
+
|
| 95 |
+
## Intended use and limitations
|
| 96 |
+
|
| 97 |
+
The checkpoint is intended for research in controlled simulation. It is not
|
| 98 |
+
intended for safety-critical control or unsupervised deployment on physical
|
| 99 |
+
robots. It may fail under visual, linguistic, dynamics, or embodiment shift.
|
| 100 |
+
Simulation performance does not establish real-world safety or robustness.
|
| 101 |
+
|
| 102 |
+
PyTorch `.pt` files use pickle-based serialization. Load only the file whose
|
| 103 |
+
checksum matches the value above, and prefer:
|
| 104 |
+
|
| 105 |
+
```python
|
| 106 |
+
torch.load(path, map_location="cpu", mmap=True, weights_only=True)
|
| 107 |
+
```
|
| 108 |
+
|
| 109 |
+
## License and attribution
|
| 110 |
+
|
| 111 |
+
The checkpoint is distributed under Apache License 2.0, consistent with the
|
| 112 |
+
Wan base model license. RIFT source code is separately distributed under the
|
| 113 |
+
license in the source repository. See [`NOTICE`](./NOTICE) for provenance and
|
| 114 |
+
attribution.
|
| 115 |
+
|
| 116 |
+
## Citation
|
| 117 |
+
|
| 118 |
+
Use the RIFT citation metadata from the matching source commit and also cite
|
| 119 |
+
the upstream [Fast-WAM paper](https://arxiv.org/abs/2603.16666).
|