Robotics
Safetensors
English
alpamayo_r1
flashdrive
autonomous-driving
vision-language-action
alpamayo
zhijianliu commited on
Commit
49b0fed
·
verified ·
1 Parent(s): 6cbd45d

Add model card

Browse files
Files changed (1) hide show
  1. README.md +76 -0
README.md ADDED
@@ -0,0 +1,76 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ pipeline_tag: robotics
3
+ library_name: transformers
4
+ license: other
5
+ language:
6
+ - en
7
+ base_model:
8
+ - nvidia/Alpamayo-R1-10B
9
+ tags:
10
+ - flashdrive
11
+ - autonomous-driving
12
+ - vision-language-action
13
+ - alpamayo
14
+ new_version: z-lab/Alpamayo-1.5-10B
15
+ ---
16
+
17
+ # Alpamayo-R1-10B (FlashDrive)
18
+
19
+ [Blog](https://z-lab.ai/projects/flashdrive/) | [GitHub](https://github.com/z-lab/flashdrive)
20
+
21
+ **FlashDrive** is an algorithm-system co-design inference framework for
22
+ [Alpamayo](https://huggingface.co/nvidia/Alpamayo-1.5-10B), NVIDIA's 10B-parameter
23
+ vision-language-action (VLA) models for autonomous driving. It combines streaming VLM
24
+ inference, [DFlash](https://github.com/z-lab/dflash) speculative reasoning, adaptive action
25
+ caching, [ParoQuant](https://github.com/z-lab/paroquant) W4A8 quantization, and system
26
+ optimizations to accelerate inference while preserving accuracy.
27
+
28
+ This repository mirrors the weights of
29
+ [nvidia/Alpamayo-R1-10B](https://huggingface.co/nvidia/Alpamayo-R1-10B) and serves as the
30
+ **base checkpoint** of the FlashDrive stack. FlashDrive derives the companion checkpoints
31
+ from this repo id by suffix:
32
+
33
+ | Repository | Contents |
34
+ |---|---|
35
+ | [z-lab/Alpamayo-R1-10B-PARO](https://huggingface.co/z-lab/Alpamayo-R1-10B-PARO) | W4A8 (ParoQuant) language-model weights |
36
+ | [z-lab/Alpamayo-R1-10B-DFlash](https://huggingface.co/z-lab/Alpamayo-R1-10B-DFlash) | DFlash block-diffusion draft model |
37
+
38
+ ## Quick Start
39
+
40
+ Install [FlashDrive](https://github.com/z-lab/flashdrive), then load the base checkpoint —
41
+ the `-PARO` and `-DFlash` companions are derived from it by suffix and fetched automatically:
42
+
43
+ ```python
44
+ from flashdrive import load_flashdrive_model
45
+
46
+ model = load_flashdrive_model("z-lab/Alpamayo-R1-10B")
47
+
48
+ pred_xyz, pred_rot = model.sample_trajectories_streaming(data)
49
+ ```
50
+
51
+ The first call per stream only prefills the KV cache and returns `(None, None)`.
52
+ For an end-to-end benchmark on a PhysicalAI-AV clip:
53
+
54
+ ```bash
55
+ python scripts/infer.py --model_path z-lab/Alpamayo-R1-10B
56
+ ```
57
+
58
+ ## Performance
59
+
60
+ FlashDrive delivers a comparable speedup on Alpamayo 1 (R1); see the [FlashDrive repository](https://github.com/z-lab/flashdrive) for benchmarks.
61
+
62
+ ## License
63
+
64
+ These weights are derived from NVIDIA's Alpamayo release and remain governed by its
65
+ [non-commercial license](https://huggingface.co/nvidia/Alpamayo-R1-10B/blob/main/LICENSE).
66
+ The [FlashDrive](https://github.com/z-lab/flashdrive) inference code is MIT.
67
+
68
+ ## Citation
69
+
70
+ ```bibtex
71
+ @article{li2026flashdrive,
72
+ title={FlashDrive: Flash Vision-Language-Action Inference For Autonomous Driving},
73
+ author={Zekai Li, Yihao Liang, Hongfei Zhang, Jian Chen, Yesheng Liang, Zhijian Liu},
74
+ year={2026}
75
+ }
76
+ ```