sihyun-yu / REPA

Official Pytorch Implementation of Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Think
https://sihyun.me/REPA
MIT License
605 stars 27 forks source link

Release SiT-XL/2 checkpoint on Hugging Face #10

Open NielsRogge opened 2 weeks ago

NielsRogge commented 2 weeks ago

Hello @sihyun-yu and team 🤗

I'm Niels and work as part of the open-source team at Hugging Face. I discovered your work on X and indexed the paper page here: https://huggingface.co/papers/2410.06940. The paper page lets people discuss about your paper and lets them find artifacts about it (your model or datasets for instance)n you can also claim the paper as yours with your HF account which will show up on your public profile.

Would you like to host the checkpoint you've pre-trained on https://huggingface.co/models? Hosting on Hugging Face will give you more visibility/enable better discoverability for people. We can add tags in the model cards so that people find the models easier, link it to the paper page, etc.

If you're down, leaving a guide here. If it's a custom PyTorch model like here, you can use PyTorchModelHubMixin class which adds from_pretrained and push_to_hub to the model which lets you to upload the model and people to download and use models right away.

If you do not want this and directly want to upload model through UI or however you want, people can also use hf_hub_download.

After uploaded, we can also link the models to the paper page (read here) so people can discover your model.

Kind regards,

Niels

sihyun-yu commented 2 weeks ago

Hi Niels - thanks for the suggestion. I will upload our checkpoint and after that will close this issue.