# PyTorchModelHubMixin Help
**URL:**
**Author:**  [@GKYagami](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Fu%2FGKYagami)%5C
**Post date:** [June 4, 2024, 11:09am UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Fpytorchmodelhubmixin-help%2F90261%2F1 "2024-06-04T11:09:11Z")
I need some help with uploading models to the Hugging Face repo to later access them with inference endpoints. I’m planning on training a custom DreamBooth model, but I’m confused about how to upload it. I guess I should be using [**PyTorchModelHubMixin**](https://huggingface.co/docs/hub/en/models-uploading), but I don’t really know how to use it in my notebook for this kind of model. Can someone please help me out?
---
**Author:**  [@nielsr](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Fu%2Fnielsr)%5C
**Post date:** [June 4, 2024, 6:14pm UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Fpytorchmodelhubmixin-help%2F90261%2F2 "2024-06-04T18:14:11Z")
Hi,
Sure happy to help out. What does your model look like?
---
**Author:**  [@GKYagami](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Fu%2FGKYagami)%5C
**Post date:** [June 4, 2024, 6:42pm UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Fpytorchmodelhubmixin-help%2F90261%2F3 "2024-06-04T18:42:48Z")
It’s basically this: [Google Colab](https://colab.research.google.com/github/ShivamShrirao/diffusers/blob/main/examples/dreambooth/DreamBooth_Stable_Diffusion.ipynb)
I managed to upload something just for testing, but I’m not really sure what to do with the above Colab notebook
---
**Author:**  [@GKYagami](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Fu%2FGKYagami)%5C
**Post date:** [June 5, 2024, 3:39pm UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Fpytorchmodelhubmixin-help%2F90261%2F4 "2024-06-05T15:39:30Z")
Hi @nielsr,
I would also like to know how to upload a ControlNet model to the hub and then send an image through the inference API as an input to the model
Control Net colab : [Google Colab](https://colab.research.google.com/github/huggingface/notebooks/blob/main/diffusers/controlnet.ipynb)
---
**Author:**  [@nielsr](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Fu%2Fnielsr)%5C
**Post date:** [June 5, 2024, 8:32pm UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Fpytorchmodelhubmixin-help%2F90261%2F5 "2024-06-05T20:32:14Z")
Hi,
As explained [here](https://huggingface.co/docs/hub/models-uploading#upload-from-a-library-with-built-in-support), if your model is available in a native library (like Diffusers or Transformers), you can use the following:
```python
from diffusers import StableDiffusionPipeline
pipe = StableDiffusionPipeline.from_pretrained(model_path, safety_checker=None, torch_dtype=torch.float16).to("cuda")
# push to hub
pipe.push_to_hub("...")
```
This is explained here: [Push files to the Hub](https://huggingface.co/docs/diffusers/en/using-diffusers/push_to_hub).
However if you have a custom nn.Module, you can leverage the [PyTorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/en/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin). I don’t see anything custom in your notebook which means that you can just use the native methods.