Add robotics pipeline tag, library name, and paper metadata
Browse filesHi! I'm Niels from the Hugging Face community science team.
I've opened this PR to update the model card with standard metadata to improve its discoverability and integration on the Hub:
- Added `pipeline_tag: robotics` to categorize the model.
- Added `library_name: transformers` as the card identifies it as the supported runtime engine.
- Linked the model to the official research paper via the `arxiv` field.
- Specified `license: other` for the custom NVIDIA NSCLv1 license.
- Added a BibTeX citation for researchers using this work.
Please review and merge if this looks good to you!
README.md
CHANGED
|
@@ -1,7 +1,19 @@
|
|
| 1 |
---
|
| 2 |
base_model:
|
| 3 |
- nvidia/Cosmos-Predict2-2B-Video2World
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 4 |
---
|
|
|
|
| 5 |
# **Cosmos-Policy-RoboCasa-Predict2-2B**
|
| 6 |
|
| 7 |
[**Cosmos Policy**](https://huggingface.co/collections/nvidia/cosmos-policy) | [**Code**](http://github.com/NVlabs/cosmos-policy) | [**White Paper**](https://arxiv.org/abs/2601.16163)| [**Website**](https://research.nvidia.com/labs/dir/cosmos-policy/)
|
|
@@ -237,4 +249,11 @@ Please report security vulnerabilities or NVIDIA AI Concerns [here](https://www.
|
|
| 237 |
|
| 238 |
If you use this model, please cite the Cosmos Policy paper:
|
| 239 |
|
| 240 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
base_model:
|
| 3 |
- nvidia/Cosmos-Predict2-2B-Video2World
|
| 4 |
+
pipeline_tag: robotics
|
| 5 |
+
library_name: transformers
|
| 6 |
+
license: other
|
| 7 |
+
tags:
|
| 8 |
+
- robotics
|
| 9 |
+
- nvidia
|
| 10 |
+
- cosmos-policy
|
| 11 |
+
- robot-manipulation
|
| 12 |
+
- visuo-motor-control
|
| 13 |
+
- imitation-learning
|
| 14 |
+
arxiv: 2601.16163
|
| 15 |
---
|
| 16 |
+
|
| 17 |
# **Cosmos-Policy-RoboCasa-Predict2-2B**
|
| 18 |
|
| 19 |
[**Cosmos Policy**](https://huggingface.co/collections/nvidia/cosmos-policy) | [**Code**](http://github.com/NVlabs/cosmos-policy) | [**White Paper**](https://arxiv.org/abs/2601.16163)| [**Website**](https://research.nvidia.com/labs/dir/cosmos-policy/)
|
|
|
|
| 249 |
|
| 250 |
If you use this model, please cite the Cosmos Policy paper:
|
| 251 |
|
| 252 |
+
```bibtex
|
| 253 |
+
@article{kim2026cosmos,
|
| 254 |
+
title={Cosmos Policy: Fine-Tuning Video Models for Visuomotor Control and Planning},
|
| 255 |
+
author={Kim, Moo Jin and Gao, Yihuai and Lin, Tsung-Yi and Lin, Yen-Chen and Ge, Yunhao and Lam, Grace and Liang, Percy and Song, Shuran and Liu, Ming-Yu and Finn, Chelsea and Gu, Jinwei},
|
| 256 |
+
journal={arXiv preprint arXiv:2601.16163},
|
| 257 |
+
year={2026}
|
| 258 |
+
}
|
| 259 |
+
```
|