File size: 5,332 Bytes
5cee0a4
8b06b32
5cee0a4
 
 
 
 
114270d
5cee0a4
 
 
 
 
f5decbe
 
 
338e0ac
f5decbe
 
 
 
5cee0a4
 
 
 
 
 
114270d
5cee0a4
f5decbe
 
 
5cee0a4
f5decbe
8b06b32
338e0ac
f5decbe
8b06b32
5cee0a4
 
f5decbe
 
5cee0a4
 
 
 
 
f5decbe
 
 
 
 
 
 
 
8808b35
f5decbe
 
8808b35
 
79e5656
8808b35
 
5cee0a4
f5decbe
5cee0a4
 
 
f5decbe
 
5cee0a4
 
 
 
 
f5decbe
 
79e5656
5cee0a4
 
f5decbe
 
 
 
 
 
 
 
5cee0a4
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
79e5656
 
 
 
 
 
 
 
f5decbe
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
---
license: cc-by-nc-4.0
language:
- en
base_model: Wan-AI/Wan2.2-TI2V-5B
base_model_relation: finetune
tags:
- arxiv:2610.25690
- robotics
- navigation
- video-diffusion
- image-text-to-action
- safetensors
library_name: scalenav
pipeline_tag: robotics
widget:
- example_title: 'ScaleNav: full navigation demo'
  output:
    url: replay.mp4
datasets:
- Yangyihui/ScaleNav-Data
---

# ScaleNav

**Scalable Internet Video Pretraining for Language-Conditioned Local Navigation**

[Paper](https://arxiv.org/abs/2610.25690) 路 [Code](https://github.com/MachEmbodied/ScaleNav) 路 [Data](https://huggingface.co/datasets/Yangyihui/ScaleNav-Data) 路 [Project page](https://zju3dv.github.io/ScaleNav/) 路 [Collection](https://huggingface.co/collections/Yangyihui/scalenav-6ad591e8ff62bb75ec75f645)

ScaleNav learns navigation priors through action-free pretraining on Internet
videos, then turns an RGB observation and a language instruction into robot actions.
It navigates to a target visible in the initial view and stops at the instructed location.

**5B parameters 路 10.02 GB checkpoint 路 [CC BY-NC 4.0](LICENSE)**

Watch the full demo in the video preview, or visit the
[project page](https://zju3dv.github.io/ScaleNav/#demos).

## Download

Complete the [installation](https://github.com/MachEmbodied/ScaleNav/blob/main/docs/INSTALL.md#environment),
then run from the ScaleNav code repository root:

```bash
python scripts/download_checkpoint.py --output checkpoints/scalenav
```

The downloader resumes interrupted transfers and verifies the pinned
[checkpoint checksums](https://github.com/MachEmbodied/ScaleNav/blob/main/docs/checkpoint-download-manifest.json).
No HF login is required for this checkpoint.

Also download the **Wan VAE, text encoder and tokenizer** (14.20 GB) following
the [checkpoint setup](https://github.com/MachEmbodied/ScaleNav/blob/main/docs/INSTALL.md#checkpoints).
For inference, use a BF16-capable NVIDIA GPU; **24 GB GPU memory and 64 GB system RAM**
are recommended.

<details>
<summary>Download with the Hugging Face CLI</summary>

```bash
hf download Yangyihui/ScaleNav --revision 8808b3589142b52e3bf0719aa0ecff8f86715a4a --local-dir checkpoints/scalenav \
  --include 'transformer/*' ar_model.pt train_meta.pt SHA256SUMS.json
```

</details>

## Inference

Use ScaleNav's `awm` CLI with your own image and an instruction describing a target
visible in it:

```bash
awm check --checkpoint checkpoints/scalenav --sha256
awm infer --checkpoint checkpoints/scalenav \
  --base-model checkpoints/Wan2.2-TI2V-5B \
  --image /path/to/observation.png \
  --prompt "Walk to the chair visible in the initial view." \
  --output outputs/example
```

Replace the image path and instruction with a matching pair. This saves nine
per-frame action deltas in `observation_trajectory.npy` and a visualization in
`observation_trajectory.png`. Actions are forward/right/clockwise-yaw displacements
in metres/metres/radians. Add `--decode-video` to save the nine predicted RGB frames.

For continuous navigation, use [`awm serve`](https://github.com/MachEmbodied/ScaleNav/blob/main/docs/INFERENCE.md#http-api).
See the [inference guide](https://github.com/MachEmbodied/ScaleNav/blob/main/docs/INFERENCE.md)
for paired examples, outputs and the robot adapter.

## Model settings

| Setting | Value |
| --- | --- |
| Observation size | 288 脳 512 |
| Observation/action rate | 4 Hz |
| Chunk length | 9 frames |
| Sampling steps | 4 |
| Video / action guidance | 5 / 1 |
| Video / action scheduler shifts | 5 / 1 |
| History | Initial-frame anchor and two recent chunks |
| Action normalization | Symlog, c = 0.01 |

## Checkpoint contents

| File | Contents |
| --- | --- |
| `transformer/diffusion_pytorch_model.safetensors` | ScaleNav model weights |
| `transformer/config.json` | Model architecture |
| `train_meta.pt`, `ar_model.pt` | Deployment metadata |
| `SHA256SUMS.json` | File checksums |

Total checkpoint size: **10.02 GB**. The separate Wan components occupy about
14.2 GB. The code repository includes [training](https://github.com/MachEmbodied/ScaleNav/blob/main/docs/TRAINING.md),
[evaluation](https://github.com/MachEmbodied/ScaleNav/blob/main/docs/EVALUATION.md)
and [trajectory visualization](https://github.com/MachEmbodied/ScaleNav/blob/main/docs/INFERENCE.md#outputs).

The released checkpoint is the final joint-fine-tuning checkpoint at step 6500.
It contains model weights and deployment metadata. Optimizer, scheduler and RNG
state for exact training resumption are not included. The Wan VAE, text encoder
and tokenizer are downloaded separately.

## License

The ScaleNav fine-tuned checkpoint is released under [CC BY-NC 4.0](LICENSE)
for noncommercial use. When sharing it or adaptations, provide attribution,
link to the license, and indicate any changes.

ScaleNav modifies [Wan2.2-TI2V-5B](https://huggingface.co/Wan-AI/Wan2.2-TI2V-5B)
through navigation pretraining, action alignment and joint fine-tuning. The
CC BY-NC 4.0 terms apply to the ScaleNav contributions; upstream Wan materials
retain their [Apache-2.0 license](LICENSE-WAN). The
[code](https://github.com/MachEmbodied/ScaleNav/blob/main/LICENSE) is separately
licensed under Apache-2.0, and [dataset components](https://huggingface.co/datasets/Yangyihui/ScaleNav-Data#sources-and-license)
retain their respective terms.