Robotics
PyTorch
robot-learning
world-action-model
flow-matching
libero
robocoin
galaxea
PoopBear commited on
Commit
f5cd093
·
verified ·
1 Parent(s): 6503cbb

docs: add RIFT model card

Browse files
Files changed (1) hide show
  1. README.md +119 -0
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).