Add robotics pipeline tag, library name, and paper metadata
#1
by
nielsr
HF Staff
- opened
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 |
+
```
|