Hi @jameslahm,

Thanks for this nice work! I wrote a quick PoC to showcase that you can easily have integration with the 🤗 hub so that you can automatically load the various YOLOv10 models using from_pretrained (and push them using push_to_hub), track download numbers for your models (similar to models in the Transformers library), and have nice model cards on a per-model basis. It leverages the PyTorchModelHubMixin class which allows to inherits these methods.

Usage is as follows:

from ultralytics import YOLOv10

model = YOLOv10.from_pretrained("nielsr/yolov10n")

# optionally, one can push a trained model to the hub
model.push_to_hub("nielsr/my-awesome-yolov10-model")

This means people don't need to manually download a checkpoint first in their local environment, it just loads automatically from the hub. The corresponding model is here for now: https://huggingface.co/nielsr/yolov10n (and the large one here: . We could move all checkpoints to separate repos to an organization on the hub or your personal user account if you're interested.

cc also @kadirnar who did a great job already at making the models available on 🤗. Would be great to leverage the PyTorchModelHubMixin from now on for new models as it makes downloads etc work :)

Would you be interested in this integration?

Kind regards,

Niels