Add robotics pipeline tag and license metadata

#1
by nielsr HF Staff - opened
Files changed (1) hide show
  1. README.md +15 -1
README.md CHANGED
@@ -2,11 +2,18 @@
2
  base_model:
3
  - nvidia/Cosmos-Policy-ALOHA-Predict2-2B
4
  - nvidia/Cosmos-Predict2-2B-Video2World
 
 
5
  ---
 
6
  # **Cosmos-Policy-ALOHA-Planning-Model-Predict2-2B**
7
 
8
  [**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/)
9
 
 
 
 
 
10
  # Model Overview
11
 
12
  ## Description:
@@ -249,4 +256,11 @@ Please report security vulnerabilities or NVIDIA AI Concerns [here](https://www.
249
 
250
  If you use this model, please cite the Cosmos Policy paper:
251
 
252
- (Cosmos Policy BibTeX citation coming soon!)
 
 
 
 
 
 
 
 
2
  base_model:
3
  - nvidia/Cosmos-Policy-ALOHA-Predict2-2B
4
  - nvidia/Cosmos-Predict2-2B-Video2World
5
+ pipeline_tag: robotics
6
+ license: other
7
  ---
8
+
9
  # **Cosmos-Policy-ALOHA-Planning-Model-Predict2-2B**
10
 
11
  [**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/)
12
 
13
+ **Paper**: [Cosmos Policy: Fine-Tuning Video Models for Visuomotor Control and Planning](https://arxiv.org/abs/2601.16163)
14
+
15
+ **Authors**: Moo Jin Kim, Yihuai Gao, Tsung-Yi Lin, Yen-Chen Lin, Yunhao Ge, Grace Lam, Percy Liang, Shuran Song, Ming-Yu Liu, Chelsea Finn, Jinwei Gu
16
+
17
  # Model Overview
18
 
19
  ## Description:
 
256
 
257
  If you use this model, please cite the Cosmos Policy paper:
258
 
259
+ ```bibtex
260
+ @article{kim2025cosmos,
261
+ title={Cosmos Policy: Fine-Tuning Video Models for Visuomotor Control and Planning},
262
+ 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},
263
+ journal={arXiv preprint arXiv:2601.16163},
264
+ year={2025}
265
+ }
266
+ ```