nielsr HF Staff commited on
Commit
e7f6db1
·
verified ·
1 Parent(s): ce4732a

Add robotics pipeline tag, library name, and paper metadata

Browse files

Hi! 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!

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
+ ```