Commit
•
f7f4ca2
1
Parent(s):
aa9579e
Push model using huggingface_hub.
Browse files
README.md
CHANGED
@@ -1,92 +1,9 @@
|
|
1 |
---
|
2 |
-
license: mit
|
3 |
-
language:
|
4 |
-
- en
|
5 |
-
pipeline_tag: robotics
|
6 |
-
library_name: diffusers
|
7 |
tags:
|
8 |
-
-
|
9 |
-
-
|
10 |
-
- pretraining
|
11 |
-
- vla
|
12 |
-
- diffusion
|
13 |
---
|
14 |
-
# RDT-1B
|
15 |
|
16 |
-
|
17 |
-
|
18 |
-
|
19 |
-
All the [code](https://github.com/GeneralEmbodiedSystem/RoboticsDiffusionTransformer/tree/main?tab=readme-ov-file) and pre-trained model weights are licensed under the MIT license.
|
20 |
-
|
21 |
-
Please refer to our [project page](https://rdt-robotics.github.io/rdt-robotics/) and [paper]() for more information.
|
22 |
-
|
23 |
-
## Model Details
|
24 |
-
|
25 |
-
- **Developed by:** The RDT team consisting of researchers from the [TSAIL group](https://ml.cs.tsinghua.edu.cn/) at Tsinghua University
|
26 |
-
- **Task Type:** Vision-Language-Action (language, image => robot actions)
|
27 |
-
- **Modle Type:** Diffusion Policy with Transformers
|
28 |
-
- **License:** MIT
|
29 |
-
- **Language(s) (NLP):** en
|
30 |
-
- **Multi-Modal Encoders:**
|
31 |
-
- **Vision Backbone:** [siglip-so400m-patch14-384](https://huggingface.co/google/siglip-so400m-patch14-384)
|
32 |
-
- **Language Model:** [t5-v1_1-xxl](https://huggingface.co/google/t5-v1_1-xxl)
|
33 |
-
- **Pre-Training Datasets:** 46 datasets consisting of [RT-1 Dataset](https://robotics-transformer1.github.io/), [RH20T](https://rh20t.github.io/), [DROID](https://droid-dataset.github.io/), [BridgeData V2](https://rail-berkeley.github.io/bridgedata/), [RoboSet](https://robopen.github.io/roboset/), and a subset of [Open X-Embodiment](https://robotics-transformer-x.github.io/). See [todo]() for a detailed list.
|
34 |
-
- **Repository:** [repo_url]
|
35 |
-
- **Paper :** [paper_url]
|
36 |
-
- **Project Page:** https://rdt-robotics.github.io/rdt-robotics/
|
37 |
-
|
38 |
-
## Uses
|
39 |
-
|
40 |
-
RDT takes language instruction, RGB image (of up to three views), control frequency (if any), and proprioception as input and predicts the next 64 robot actions in the form of the unified action space vector.
|
41 |
-
The unified action space vector includes all the main physical quantities of the robot manipulator (e.g., the end-effector and joint, position and velocity, and base movement).
|
42 |
-
To deploy on your robot platform, you need to pick the relevant quantities from the unified vector. See our repository for more information.
|
43 |
-
|
44 |
-
**Out-of-Scope**: Due to the embodiment gap, RDT cannot yet generalize to new robot platforms (not seen in the pre-training datasets).
|
45 |
-
In this case, we recommend collecting a small dataset of the target robot and then using it to fine-tune RDT.
|
46 |
-
See our repository for a tutorial.
|
47 |
-
|
48 |
-
Here's an example of how to use the RDT-1B model for inference on a robot:
|
49 |
-
```python
|
50 |
-
# Clone the repository and install dependencies
|
51 |
-
from scripts.agilex_model import create_model
|
52 |
-
# Names of cameras used for visual input
|
53 |
-
CAMERA_NAMES = ['cam_high', 'cam_right_wrist', 'cam_left_wrist']
|
54 |
-
config = {
|
55 |
-
'episode_len': 1000, # Max length of one episode
|
56 |
-
'state_dim': 14, # Dimension of the robot's state
|
57 |
-
'chunk_size': 64, # Number of actions to predict in one step
|
58 |
-
'camera_names': CAMERA_NAMES,
|
59 |
-
}
|
60 |
-
pretrained_vision_encoder_name_or_path = "google/siglip-so400m-patch14-384"
|
61 |
-
# Create the model with the specified configuration
|
62 |
-
model = create_model(
|
63 |
-
args=config,
|
64 |
-
dtype=torch.bfloat16,
|
65 |
-
pretrained_vision_encoder_name_or_path=pretrained_vision_encoder_name_or_path,
|
66 |
-
control_frequency=25,
|
67 |
-
)
|
68 |
-
# Start inference process
|
69 |
-
# Load pre-computed language embeddings
|
70 |
-
lang_embeddings_path = 'your/language/embedding/path'
|
71 |
-
text_embedding = torch.load(lang_embeddings_path)['embeddings']
|
72 |
-
images: List(PIL.Image) = ... # The images from last 2 frame
|
73 |
-
proprio = ... # The current robot state
|
74 |
-
# Perform inference to predict the next chunk_size actions
|
75 |
-
actions = policy.step(
|
76 |
-
proprio=proprio,
|
77 |
-
images=images,
|
78 |
-
text_embeds=text_embedding
|
79 |
-
)
|
80 |
-
```
|
81 |
-
|
82 |
-
<!-- RDT-1B supports finetuning on custom datasets, deploying and inferencing on real robots, as well as retraining the model.
|
83 |
-
Please refer to [our repository](https://github.com/GeneralEmbodiedSystem/RoboticsDiffusionTransformer/blob/main/docs/pretrain.md) for all the above guides. -->
|
84 |
-
|
85 |
-
|
86 |
-
## Citation
|
87 |
-
|
88 |
-
<!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
|
89 |
-
|
90 |
-
**BibTeX:**
|
91 |
-
|
92 |
-
[More Information Needed]
|
|
|
1 |
---
|
|
|
|
|
|
|
|
|
|
|
2 |
tags:
|
3 |
+
- pytorch_model_hub_mixin
|
4 |
+
- model_hub_mixin
|
|
|
|
|
|
|
5 |
---
|
|
|
6 |
|
7 |
+
This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
|
8 |
+
- Library: https://huggingface.co/robotics-diffusion-transformer/rdt-1b
|
9 |
+
- Docs: [More Information Needed]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|