text
If you're a researcher or developing your own model 👀 you might need to take a look at huggingface's ModelHubMixin classes.
If you're a researcher or developing your own model 👀 you might need to take a look at huggingface's ModelHubMixin classes.
new_line
text
They are used to seamlessly integrate your AI model with huggingface and to save/ load your model easily 🚀
They are used to seamlessly integrate your AI model with huggingface and to save/ load your model easily 🚀
new_line
new_line
text
1️⃣ make sure you're using the appropriate library version
1️⃣ make sure you're using the appropriate library version
new_line
code_fence
```
pip install -qU "huggingface_hub>=0.22"
```
pip install -qU "huggingface_hub>=0.22"
new_line
text
2️⃣  inherit from the appropriate class
2️⃣  inherit from the appropriate class
new_line
code_fence
```python
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)
```
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)
python
new_line
text
4️⃣ push the model to the hub (or use save_pretrained method to save locally) 
4️⃣ push the model to the hub (or use save_pretrained method to save locally) 
new_line
code_fence
```python
first_model.push_to_hub("not-lain/test")
```
first_model.push_to_hub("not-lain/test")
python
new_line
text
5️⃣  Load and initialize the model from the hub using the original class
5️⃣  Load and initialize the model from the hub using the original class
new_line
code_fence
```python
pretrained_model = MyModel.from_pretrained("not-lain/test")
```
pretrained_model = MyModel.from_pretrained("not-lain/test")
python
new_line
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
```
pip install -qU "huggingface_hub>=0.22"
```
2️⃣  inherit from the appropriate class
```python
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) 
```python
first_model.push_to_hub("not-lain/test")
```
5️⃣  Load and initialize the model from the hub using the original class
```python
pretrained_model = MyModel.from_pretrained("not-lain/test")
```

👍
DmitryRyumin KvrParaskevi DynaTechSystems osanseviero isidentical cc92yy3344 KingNish Wauplin Ramikan-BR
🔥
KvrParaskevi osanseviero louisbrulenaudet alielfilali01 KingNish Wauplin Ramikan-BR