Add robotics pipeline tag, library name, and paper metadata

#1
by nielsr HF Staff - opened
Files changed (1) hide show
  1. README.md +20 -1
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
- (Cosmos Policy BibTeX citation coming soon!)
 
 
 
 
 
 
 
 
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
+ ```