cosmos-policy

Add robotics pipeline tag and license metadata

#1
by nielsr HF Staff - opened
Files changed (1) hide show
  1. README.md +14 -4
README.md CHANGED
@@ -1,9 +1,12 @@
1
  ---
2
- datasets:
3
- - nvidia/ALOHA-Cosmos-Policy
4
  base_model:
5
  - nvidia/Cosmos-Predict2-2B-Video2World
 
 
 
 
6
  ---
 
7
  # **Cosmos-Policy-ALOHA-Predict2-2B**
8
 
9
  [**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/)
@@ -258,11 +261,18 @@ Please report security vulnerabilities or NVIDIA AI Concerns [here](https://www.
258
  - **Base Model**: [Cosmos-Predict2-2B-Video2World](https://huggingface.co/nvidia/Cosmos-Predict2-2B-Video2World)
259
  - **Training Dataset**: [ALOHA-Cosmos-Policy](https://huggingface.co/datasets/nvidia/ALOHA-Cosmos-Policy)
260
  - **Planning Model Checkpoint**: [Cosmos-Policy-ALOHA-Planning-Model-Predict2-2B](https://huggingface.co/nvidia/Cosmos-Policy-ALOHA-Planning-Model-Predict2-2B)
261
- - **Paper**: https://arxiv.org/abs/2601.16163
262
  - **Original ALOHA**: [Learning Fine-Grained Bimanual Manipulation with Low-Cost Hardware](https://arxiv.org/abs/2304.13705)
263
 
264
  ## Citation
265
 
266
  If you use this model, please cite the Cosmos Policy paper:
267
 
268
- (Cosmos Policy BibTeX citation coming soon!)
 
 
 
 
 
 
 
 
1
  ---
 
 
2
  base_model:
3
  - nvidia/Cosmos-Predict2-2B-Video2World
4
+ datasets:
5
+ - nvidia/ALOHA-Cosmos-Policy
6
+ license: other
7
+ pipeline_tag: robotics
8
  ---
9
+
10
  # **Cosmos-Policy-ALOHA-Predict2-2B**
11
 
12
  [**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/)
 
261
  - **Base Model**: [Cosmos-Predict2-2B-Video2World](https://huggingface.co/nvidia/Cosmos-Predict2-2B-Video2World)
262
  - **Training Dataset**: [ALOHA-Cosmos-Policy](https://huggingface.co/datasets/nvidia/ALOHA-Cosmos-Policy)
263
  - **Planning Model Checkpoint**: [Cosmos-Policy-ALOHA-Planning-Model-Predict2-2B](https://huggingface.co/nvidia/Cosmos-Policy-ALOHA-Planning-Model-Predict2-2B)
264
+ - **Paper**: [Cosmos Policy: Fine-Tuning Video Models for Visuomotor Control and Planning](https://arxiv.org/abs/2601.16163)
265
  - **Original ALOHA**: [Learning Fine-Grained Bimanual Manipulation with Low-Cost Hardware](https://arxiv.org/abs/2304.13705)
266
 
267
  ## Citation
268
 
269
  If you use this model, please cite the Cosmos Policy paper:
270
 
271
+ ```bibtex
272
+ @article{kim2026cosmos,
273
+ title={Cosmos Policy: Fine-Tuning Video Models for Visuomotor Control and Planning},
274
+ 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},
275
+ journal={arXiv preprint arXiv:2601.16163},
276
+ year={2026}
277
+ }
278
+ ```