Release ChatDiT on Hugging Face

[NielsRogge]

Hello @lhhuang 🤗 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 as yours got featured: https://huggingface.co/papers/2412.12571. 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. Looking forward to the code and model release. Would you like to host the model you've pre-trained on https://huggingface.co/models? 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. If you're down, leaving a guide here. If it's a custom PyTorch model, you can use the 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. You can also build a demo for your model on Spaces, we can provide you an A100 grant. Let me know if you're interested/need any guidance :) Kind regards, Niels