Files
diffusers/docs/source/en/api/models/overview.md
M. Tolga Cangöz a359ff7644 [Docs] Fix typos and update files at API's Main Classes, Models, and Schedulers pages (#5720)
* Fix typos, update, add Copyright info, and trim trailing whitespaces

* Update docs/source/en/api/loaders.md

Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>

* Update docs/source/en/api/models/autoencoder_tiny.md

Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>

* Update docs/source/en/api/models/autoencoder_tiny.md

Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>

---------

Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>
2023-11-13 14:32:59 -08:00

1.2 KiB

Models

🤗 Diffusers provides pretrained models for popular algorithms and modules to create custom diffusion systems. The primary function of models is to denoise an input sample as modeled by the distribution \(p_{\theta}(x_{t-1}|x_{t})\).

All models are built from the base [ModelMixin] class which is a torch.nn.Module providing basic functionality for saving and loading models, locally and from the Hugging Face Hub.

ModelMixin

autodoc ModelMixin

FlaxModelMixin

autodoc FlaxModelMixin

PushToHubMixin

autodoc utils.PushToHubMixin