a template for the PyTorchModelHubMixin
pip install git+https://github.com/not-lain/PyTorchModelHubMixin-template.git
from MyModule import MyModel # already added to src\MyModule\__init__.py
model = MyModel(a=5,b=3) # initialize the model
model.save_pretrained(local_path) # save locally
model.push_to_hub(repo_id) # push to 🤗from MyModule import MyModel
model = MyModel.from_pretrained(path_or_repo_id) # initialize the model and inject the weightsafter bilding your own library (this repo is considered a library) you can open a pull request on this file to make your library semi-official on huggingface