Leverage PyTorchModelHubMixin
#1
by
nielsr
HF staff
- opened
Hi @kadirnar ,
Thanks so much for uploading YOLOv10 checkpoints to make them easier discoverable. :)
Small note though, we prefer to have each model checkpoint uploaded to a separate repository. With the mixin class, it allows to add from_pretrained
and push_to_hub
capabilities to each model, download stats, automatic model card, etc.
See the docs here: https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin
Opened a PR here: https://github.com/THU-MIG/yolov10/pull/168