FengheTan9
Release MambaMIM on Hugging Face
Hi @FengheTan9 🤗
Niels here from the open-source team at Hugging Face. I discovered your work on Arxiv and was wondering whether you would like to submit it to hf.co/papers to improve its discoverability.If you are one of the authors, you can submit it at https://huggingface.co/papers/submit.
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'd be great to make the MambaMIM checkpoint available on the 🤗 hub, to improve its discoverability/visibility. We can add tags so that people find them when filtering https://huggingface.co/models and https://huggingface.co/datasets.
Also, awesome that you have released the weights, but hosting them on Google Drive makes it a bit harder to integrate into the ecosystem (e.g. you can't do from_pretrained).
Would you like to also release it on https://huggingface.co/models, instead of Google Drive?
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!
Cheers,
Niels ML Engineer @ HF 🤗
Awesome! Could you link it in this Github repo?
Btw as an FYI, this is how people can download it:
from huggingface_hub import hf_hub_download
filepath = hf_hub_download(repo_id="FengheTan9/MambaMIM", filename="mambamim_mask75.pth")
Hence if you're interested, it could make sense to integrate hf_hub_download in your code base so as to automatically download the checkpoint from the hub. See here for a guide on how people typically add a from_pretrained method to their class: https://huggingface.co/docs/huggingface_hub/guides/integrations#frompretrained