Post
1527
If you're a researcher or developing your own model 👀 you might need to take a look at huggingface's ModelHubMixin classes.
They are used to seamlessly integrate your AI model with huggingface and to save/ load your model easily 🚀
1️⃣ make sure you're using the appropriate library version
2️⃣ inherit from the appropriate class
4️⃣ push the model to the hub (or use save_pretrained method to save locally)
5️⃣ Load and initialize the model from the hub using the original class
They are used to seamlessly integrate your AI model with huggingface and to save/ load your model easily 🚀
1️⃣ make sure you're using the appropriate library version
pip install -qU "huggingface_hub>=0.22"2️⃣ inherit from the appropriate class
from huggingface_hub import PyTorchModelHubMixin
from torch import nn
class MyModel(nn.Module,PyTorchModelHubMixin):
def __init__(self, a, b):
super().__init__()
self.layer = nn.Linear(a,b)
def forward(self,inputs):
return self.layer(inputs)
first_model = MyModel(3,1)4️⃣ push the model to the hub (or use save_pretrained method to save locally)
first_model.push_to_hub("not-lain/test")5️⃣ Load and initialize the model from the hub using the original class
pretrained_model = MyModel.from_pretrained("not-lain/test")