Improve model card: Add pipeline tag, paper link, project page, and code link (#1)
Browse files- Improve model card: Add pipeline tag, paper link, project page, and code link (7a052d17f4ec15a0d1b4ad0ecb6390f0378bf21b)
Co-authored-by: Niels Rogge <[email protected]>
README.md
CHANGED
|
@@ -1,91 +1,76 @@
|
|
| 1 |
-
---
|
| 2 |
-
license: apache-2.0
|
| 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 |
-
title={PPTmodel4UnitreeG1: Post-Pre-trained Model for UnitreeG1 Robot Applications},
|
| 78 |
-
author={l2aggle},
|
| 79 |
-
year={2024},
|
| 80 |
-
url={https://huggingface.co/l2aggle/PPTmodel4UnitreeG1}
|
| 81 |
-
}
|
| 82 |
-
```
|
| 83 |
-
|
| 84 |
-
## Related Resources
|
| 85 |
-
|
| 86 |
-
- Dataset: [Agibot2UnitreeG1Retarget](https://huggingface.co/datasets/l2aggle/Agibot2UnitreeG1Retarget)
|
| 87 |
-
- Model Repository: [PPTmodel4UnitreeG1](https://huggingface.co/l2aggle/PPTmodel4UnitreeG1)
|
| 88 |
-
|
| 89 |
-
## Contact
|
| 90 |
-
|
| 91 |
-
For questions or issues regarding this model, please open an issue in the model repository or contact the author through Hugging Face. -->
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: apache-2.0
|
| 3 |
+
pipeline_tag: robotics
|
| 4 |
+
library_name: transformers
|
| 5 |
+
---
|
| 6 |
+
|
| 7 |
+
# PPTmodel4UnitreeG1
|
| 8 |
+
|
| 9 |
+
This is the PPTmodel4UnitreeG1 model presented in [TrajBooster: Boosting Humanoid Whole-Body Manipulation via Trajectory-Centric Learning](https://huggingface.co/papers/2509.11839).
|
| 10 |
+
|
| 11 |
+
Project page: https://jiachengliu3.github.io/TrajBooster/
|
| 12 |
+
Code: https://github.com/jiachengliu3/OpenTrajBooster
|
| 13 |
+
|
| 14 |
+
This model is a post-pre-trained model specifically designed for Unitree G1 robot applications. The model has been fine-tuned using the [Agibot2UnitreeG1Retarget dataset](https://huggingface.co/datasets/l2aggle/Agibot2UnitreeG1Retarget) to enhance its performance on robotic whole-body manipulation.
|
| 15 |
+
|
| 16 |
+
## Model Description
|
| 17 |
+
|
| 18 |
+
This model underwent post-pre-training using specialized robotics data to improve its understanding and generation capabilities for Unitree G1 humanoid robot applications. The training process leveraged the Agibot2UnitreeG1Retarget dataset, which contains motion retargeting data specifically curated for Unitree G1.
|
| 19 |
+
|
| 20 |
+
## Dataset
|
| 21 |
+
|
| 22 |
+
The model was trained on the [Agibot2UnitreeG1Retarget dataset](https://huggingface.co/datasets/l2aggle/Agibot2UnitreeG1Retarget), which provides comprehensive motion retargeting data for converting motion patterns to UnitreeG1 robot format.
|
| 23 |
+
|
| 24 |
+
## Model Files
|
| 25 |
+
|
| 26 |
+
The model consists of the following files:
|
| 27 |
+
- `config.json` - Model configuration
|
| 28 |
+
- `model.safetensors.index.json` - SafeTensors index file
|
| 29 |
+
- `model-00001-of-00002.safetensors` - Model weights (part 1)
|
| 30 |
+
- `model-00002-of-00002.safetensors` - Model weights (part 2)
|
| 31 |
+
- `trainer_state.json` - Training state information
|
| 32 |
+
- `training_args.bin` - Training arguments
|
| 33 |
+
- `experiment_cfg/` - Experimental configuration files
|
| 34 |
+
|
| 35 |
+
## Download and Usage
|
| 36 |
+
|
| 37 |
+
### Method 1: Using Hugging Face Hub
|
| 38 |
+
|
| 39 |
+
```python
|
| 40 |
+
from transformers import AutoModel, AutoTokenizer
|
| 41 |
+
|
| 42 |
+
# Download and load the model
|
| 43 |
+
model = AutoModel.from_pretrained("l2aggle/PPTmodel4UnitreeG1")
|
| 44 |
+
tokenizer = AutoTokenizer.from_pretrained("l2aggle/PPTmodel4UnitreeG1")
|
| 45 |
+
```
|
| 46 |
+
|
| 47 |
+
### Method 2: Using Git LFS
|
| 48 |
+
|
| 49 |
+
```bash
|
| 50 |
+
# Clone the repository
|
| 51 |
+
git clone https://huggingface.co/l2aggle/PPTmodel4UnitreeG1
|
| 52 |
+
|
| 53 |
+
# Navigate to the model directory
|
| 54 |
+
cd PPTmodel4UnitreeG1
|
| 55 |
+
```
|
| 56 |
+
|
| 57 |
+
### Method 3: Direct Download
|
| 58 |
+
|
| 59 |
+
You can also download individual files directly from the [model repository](https://huggingface.co/l2aggle/PPTmodel4UnitreeG1/tree/main) on Hugging Face.
|
| 60 |
+
|
| 61 |
+
## Requirements
|
| 62 |
+
|
| 63 |
+
- Python 3.7+
|
| 64 |
+
- PyTorch
|
| 65 |
+
- Transformers library
|
| 66 |
+
- SafeTensors
|
| 67 |
+
|
| 68 |
+
## Installation
|
| 69 |
+
|
| 70 |
+
```bash
|
| 71 |
+
pip install torch transformers safetensors
|
| 72 |
+
```
|
| 73 |
+
|
| 74 |
+
## License
|
| 75 |
+
|
| 76 |
+
This model is released under the Apache 2.0 license.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|