Improve model card: add pipeline tag, sample usage, abstract, teaser image, and correct license

#1
by nielsr HF Staff - opened
Files changed (1) hide show
  1. README.md +112 -9
README.md CHANGED
@@ -1,15 +1,24 @@
1
  ---
2
- license: cc-by-nc-sa-4.0
 
3
  tags:
4
- - robotics
5
  - vision-language-action-model
6
  - vision-language-model
7
  ---
 
8
  # Model Card for InternVLA-M1_object
9
- InternVLA-M1 is an open-source, end-to-end vision–language–action (VLA) framework for building and researching generalist robot policies.
 
10
  - 🌐 Homepage: [InternVLA-M1 Project Page](https://internrobotics.github.io/internvla-m1.github.io/)
11
  - 💻 Codebase: [InternVLA-M1 GitHub Repo](https://github.com/InternRobotics/InternVLA-M1)
12
 
 
 
 
 
 
 
 
13
  ## Training Details
14
  ```
15
  action_chunk: 8
@@ -17,12 +26,106 @@ batch_size: 128
17
  training_steps: 30k
18
  ```
19
 
20
- ## Citation
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
21
  ```
22
- @misc{internvla2024,
23
- title = {InternVLA-M1: A Spatially Guided Vision-Language-Action Framework for Generalist Robot Policy},
24
- author = {InternVLA-M1 Contributors},
25
- year = {2025},
26
- booktitle={arXiv},
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
27
  }
28
  ```
 
1
  ---
2
+ license: mit
3
+ pipeline_tag: robotics
4
  tags:
 
5
  - vision-language-action-model
6
  - vision-language-model
7
  ---
8
+
9
  # Model Card for InternVLA-M1_object
10
+ InternVLA-M1 is an open-source, end-to-end vision–language–action (VLA) framework for building and researching generalist robot policies, as introduced in the paper [InternVLA-M1: A Spatially Guided Vision-Language-Action Framework for Generalist Robot Policy](https://huggingface.co/papers/2510.13778).
11
+
12
  - 🌐 Homepage: [InternVLA-M1 Project Page](https://internrobotics.github.io/internvla-m1.github.io/)
13
  - 💻 Codebase: [InternVLA-M1 GitHub Repo](https://github.com/InternRobotics/InternVLA-M1)
14
 
15
+ <div align="center">
16
+ <img src="https://raw.githubusercontent.com/InternRobotics/InternVLA-M1/main/assets/teaser.png" width="100%" height="100%"/>
17
+ </div>
18
+
19
+ ## Abstract
20
+ We introduce InternVLA-M1, a unified framework for spatial grounding and robot control that advances instruction-following robots toward scalable, general-purpose intelligence. Its core idea is spatially guided vision-language-action training, where spatial grounding serves as the critical link between instructions and robot actions. InternVLA-M1 employs a two-stage pipeline: (i) spatial grounding pre-training on over 2.3M spatial reasoning data to determine "where to act" by aligning instructions with visual, embodiment-agnostic positions, and (ii) spatially guided action post-training to decide "how to act" by generating embodiment-aware actions through plug-and-play spatial prompting. This spatially guided training recipe yields consistent gains: InternVLA-M1 outperforms its variant without spatial guidance by +14.6% on SimplerEnv Google Robot, +17% on WidowX, and +4.3% on LIBERO Franka, while demonstrating stronger spatial reasoning capability in box, point, and trace prediction. To further scale instruction following, we built a simulation engine to collect 244K generalizable pick-and-place episodes, enabling a 6.2% average improvement across 200 tasks and 3K+ objects. In real-world clustered pick-and-place, InternVLA-M1 improved by 7.3%, and with synthetic co-training, achieved +20.6% on unseen objects and novel configurations. Moreover, in long-horizon reasoning-intensive scenarios, it surpassed existing works by over 10%. These results highlight spatially guided training as a unifying principle for scalable and resilient generalist robots.
21
+
22
  ## Training Details
23
  ```
24
  action_chunk: 8
 
26
  training_steps: 30k
27
  ```
28
 
29
+ ## Sample Usage
30
+
31
+ Below are two collapsible examples: InternVLA-M1 chat and action prediction.
32
+
33
+ <details open>
34
+ <summary><b>InternVLA-M1 Chat Demo (image Q&A / Spatial Grounding)</b></summary>
35
+
36
+ ```python
37
+ from InternVLA.model.framework.M1 import InternVLA_M1
38
+ from PIL import Image
39
+ import requests
40
+ from io import BytesIO
41
+ import torch
42
+
43
+ def load_image_from_url(url: str) -> Image.Image:
44
+ resp = requests.get(url, timeout=15)
45
+ resp.raise_for_status()
46
+ img = Image.open(BytesIO(resp.content)).convert("RGB")
47
+ return img
48
+
49
+ saved_model_path = "/PATH/checkpoints/steps_50000_pytorch_model.pt"
50
+ internVLA_M1 = InternVLA_M1.from_pretrained(saved_model_path)
51
+
52
+ # Use the raw image link for direct download
53
+ image_url = "https://raw.githubusercontent.com/InternRobotics/InternVLA-M1/InternVLA-M1/assets/table.jpeg"
54
+ image = load_image_from_url(image_url)
55
+ question = "Give the bounding box for the apple."
56
+ response = internVLA_M1.chat_with_M1(image, question)
57
+ print(response)
58
+ ```
59
+ </details>
60
+
61
+ <details>
62
+ <summary><b>InternVLA-M1 Action Prediction Demo (two views)</b></summary>
63
+
64
+ ```python
65
+ from InternVLA.model.framework.M1 import InternVLA_M1
66
+ from PIL import Image
67
+ import requests
68
+ from io import BytesIO
69
+ import torch
70
+
71
+ def load_image_from_url(url: str) -> Image.Image:
72
+ resp = requests.get(url, timeout=15)
73
+ resp.raise_for_status()
74
+ img = Image.open(BytesIO(resp.content)).convert("RGB")
75
+ return img
76
+
77
+ saved_model_path = "/PATH/checkpoints/steps_50000_pytorch_model.pt"
78
+ internVLA_M1 = InternVLA_M1.from_pretrained(saved_model_path)
79
+
80
+ image_url = "https://raw.githubusercontent.com/InternRobotics/InternVLA-M1/InternVLA-M1/assets/table.jpeg"
81
+ view1 = load_image_from_url(image_url)
82
+ view2 = view1.copy()
83
+
84
+ # Construct input: batch size = 1, two views
85
+ batch_images = [[view1, view2]] # List[List[PIL.Image]]
86
+ instructions = ["Pick up the apple and place it on the plate."]
87
+
88
+ if torch.cuda.is_available():
89
+ internVLA_M1 = internVLA_M1.to("cuda")
90
+
91
+ pred = internVLA_M1.predict_action(
92
+ batch_images=batch_images,
93
+ instructions=instructions,
94
+ cfg_scale=1.5,
95
+ use_ddim=True,
96
+ num_ddim_steps=10,
97
+ )
98
+ normalized_actions = pred["normalized_actions"] # [B, T, action_dim]
99
+ print(normalized_actions.shape, type(normalized_actions))
100
  ```
101
+ </details>
102
+
103
+ ## Acknowledgements
104
+
105
+ We thank the open-source community for their inspiring work. This project builds upon and is inspired by the following projects (alphabetical order):
106
+ - [IPEC-COMMUNITY](https://huggingface.co/IPEC-COMMUNITY): Curated OXE / LIBERO style multi-task datasets and formatting examples.
107
+ - [Isaac-GR00T](https://github.com/NVIDIA/Isaac-GR00T): Standardized action data loader (GR00T-LeRobot).
108
+ - [Qwen2.5-VL](https://github.com/QwenLM/Qwen2.5-VL/blob/main/qwen-vl-finetune/README.md): Multimodal input/output format, data loader, and pretrained VLM backbone.
109
+ - [CogACT](https://github.com/microsoft/CogACT/tree/main/action_model): Reference for a DiT-style action head design.
110
+ - [Llavavla](https://github.com/JinhuiYE/llavavla): Baseline code structure and engineering design references.
111
+ - [GenManip Simulation Platform](https://github.com/InternRobotics/GenManip): Simulation platform for generalizable pick-and-place based on Isaac Sim.
112
+
113
+ Notes:
114
+ - If any required attribution or license header is missing, please open an issue and we will correct it promptly.
115
+ - All third-party resources remain under their original licenses; users should comply with respective terms.
116
+
117
+ ---
118
+
119
+ Thanks for using **InternVLA-M1**! 🌟
120
+ If you find it useful, please consider giving us a ⭐ on GitHub.
121
+
122
+ ## Citation
123
+ ```bibtex
124
+ @article{internvlam1,
125
+ title = {InternVLA-M1: A Spatially Guided Vision-Language-Action Framework for Generalist Robot Policy},
126
+ author = {InternVLA-M1 Contributors},
127
+ journal = {arXiv preprint arXiv:2510.13778},
128
+ year = {2025},
129
+ url = {https://huggingface.co/papers/2510.13778}
130
  }
131
  ```