alexnwang commited on
Commit
441101d
·
0 Parent(s):

Release checkpoints + model card

Browse files
.gitattributes ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ *.7z filter=lfs diff=lfs merge=lfs -text
2
+ *.arrow filter=lfs diff=lfs merge=lfs -text
3
+ *.bin filter=lfs diff=lfs merge=lfs -text
4
+ *.bz2 filter=lfs diff=lfs merge=lfs -text
5
+ *.ckpt filter=lfs diff=lfs merge=lfs -text
6
+ *.ftz filter=lfs diff=lfs merge=lfs -text
7
+ *.gz filter=lfs diff=lfs merge=lfs -text
8
+ *.h5 filter=lfs diff=lfs merge=lfs -text
9
+ *.joblib filter=lfs diff=lfs merge=lfs -text
10
+ *.lfs.* filter=lfs diff=lfs merge=lfs -text
11
+ *.mlmodel filter=lfs diff=lfs merge=lfs -text
12
+ *.model filter=lfs diff=lfs merge=lfs -text
13
+ *.msgpack filter=lfs diff=lfs merge=lfs -text
14
+ *.npy filter=lfs diff=lfs merge=lfs -text
15
+ *.npz filter=lfs diff=lfs merge=lfs -text
16
+ *.onnx filter=lfs diff=lfs merge=lfs -text
17
+ *.ot filter=lfs diff=lfs merge=lfs -text
18
+ *.parquet filter=lfs diff=lfs merge=lfs -text
19
+ *.pb filter=lfs diff=lfs merge=lfs -text
20
+ *.pickle filter=lfs diff=lfs merge=lfs -text
21
+ *.pkl filter=lfs diff=lfs merge=lfs -text
22
+ *.pt filter=lfs diff=lfs merge=lfs -text
23
+ *.pth filter=lfs diff=lfs merge=lfs -text
24
+ *.rar filter=lfs diff=lfs merge=lfs -text
25
+ *.safetensors filter=lfs diff=lfs merge=lfs -text
26
+ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
27
+ *.tar.* filter=lfs diff=lfs merge=lfs -text
28
+ *.tar filter=lfs diff=lfs merge=lfs -text
29
+ *.tflite filter=lfs diff=lfs merge=lfs -text
30
+ *.tgz filter=lfs diff=lfs merge=lfs -text
31
+ *.wasm filter=lfs diff=lfs merge=lfs -text
32
+ *.xz filter=lfs diff=lfs merge=lfs -text
33
+ *.zip filter=lfs diff=lfs merge=lfs -text
34
+ *.zst filter=lfs diff=lfs merge=lfs -text
35
+ *tfevents* filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ library_name: pytorch
4
+ tags:
5
+ - robotics
6
+ - navigation
7
+ - world-model
8
+ - diffusion-policy
9
+ - nymeria
10
+ ---
11
+
12
+ # Lifting Embodied World Models for Planning and Control — checkpoints
13
+
14
+ Pretrained checkpoints for **[Lifting Embodied World Models for Planning and Control](https://www.alexn.wang/lwm/)**.
15
+
16
+ - **Code:** https://github.com/alexnwang/lifted-world-model
17
+ - **Paper:** https://arxiv.org/abs/2604.26182
18
+ - This repo also contains a checkpoint for [PEVA](https://dannytran123.github.io/PEVA/).
19
+
20
+ ## Contents
21
+
22
+ | Path | What it is |
23
+ |---|---|
24
+ | `waypoint_policy/ema_9.pth` | Diffusion waypoint policy weights (Nymeria, `goal_type: draw`). The architecture is read from the in-repo `config/waypoint_policy.yaml`. |
25
+ | `peva/nymeria_peva.pth.tar` | PEVA video world model (Nymeria) used by the CEM planner. |
26
+
27
+ ## Usage
28
+
29
+ See the [repo README](https://github.com/alexnwang/lifted-world-model) for full setup. In brief:
30
+
31
+ ```bash
32
+ python plan_cem.py \
33
+ --model_dir <DOWNLOAD>/waypoint_policy \
34
+ --peva_checkpoint <DOWNLOAD>/peva/nymeria_peva.pth.tar \
35
+ --data_folder <NYMERIA_DATA_ROOT>
36
+ ```
peva/nymeria_peva.pth.tar ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9fbb83cb89692fc0eaa118dc66516a53bf10e62765140fc6bba499a8aa4b184d
3
+ size 15623932640
waypoint_policy/ema_9.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:df5baf8df7b306684b86e8ef23fbaef484b97a42bd23992527fd70f71ad4cbcb
3
+ size 234152711