Release RoboMatrix model weights on Hugging Face

[NielsRogge]

Hi @WayneMao 🤗

I'm Niels and work as part of the open-source team at Hugging Face. I discovered your work through Hugging Face's daily papers,
the paper page is here: https://huggingface.co/papers/2412.00171.
The paper page lets people discuss about your paper and lets them find artifacts about it (your models for instance),
you can also claim the paper as yours which will show up on your public profile at HF, add Github and project page URLs.

It's great to see the RoboMatrix dataset being released on Hugging Face. Would you like to also host the model weights on Hugging Face?
Hosting on Hugging Face will give you more visibility/enable better discoverability. We can add tags in the model cards so that people find
the models easier, link it to the paper page, etc.

We can add tags so that people find them when filtering https://huggingface.co/models.

See here for a guide: https://huggingface.co/docs/hub/models-uploading.

In this case, we could leverage the PyTorchModelHubMixin class which adds from_pretrained and push_to_hub to any custom nn.Module. Alternatively, one can leverages the hf_hub_download one-liner to download a checkpoint from the hub.

We encourage researchers to push each model checkpoint to a separate model repository, so that things like download stats also work. We can then also link the checkpoints to the paper page.

Let me know if you're interested/need any help regarding this!

Kind regards,

Niels