search

octree-nn / octfusion

OctFusion: Octree-based Diffusion Models for 3D Shape Generation
143 stars 4 forks source link

Host models on HF #4

Open NielsRogge opened 2 months ago

NielsRogge commented 2 months ago

Hi,

Niels here from the open-source team at Hugging Face. I discovered your work through the paper page: https://huggingface.co/papers/2408.14732 (feel free to claim the paper so that it appears under your HF account!).

I see the checkpoints are currently made available on Google Drive. To improve the discoverability of your work, it would be great to make the models available on the 🤗 hub. We can also add tags to the model cards such as "image-to-3d" so that people can find them at https://huggingface.co/models?pipeline_tag=image-to-3d.

See here for a guide: https://huggingface.co/docs/hub/models-uploading. In this case, the PyTorchModelHubMixin might be useful given the model is a custom PyTorch model.

Alternatively, one could also use the web interface and use the handy 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.

Let me know if you need any help regarding this!

Cheers,

Niels ML Engineer @ HF 🤗

ymxbj commented 2 months ago

Thanks for your attention. It is indeed a useful suggestion. We are considering making the models available on HuggingFace to improve the discoverability of our work.