class PyTorchModelHubMixin(ModelHubMixin):
(...)

def _save_pretrained(self, save_directory: Path) -> None:
"""Save weights from a Pytorch model to a local directory."""
save_model_as_safetensor(self.module, str(save_directory / SAFETENSORS_SINGLE_FILE))

```  
3. Implement the `_from_pretrained` method:  
```python
class PyTorchModelHubMixin(ModelHubMixin):
(...)