Files
diffusers/docs/source/en/api/loaders/single_file.md
Steven Liu 7457aa67cb [docs] Loader APIs (#5813)
* first draft

* remove old loader doc

* start adding lora code examples

* finish

* add link to loralinearlayer

* feedback

* fix
2023-11-20 10:53:13 -08:00

1.7 KiB

Single files

Diffusers supports loading pretrained pipeline (or model) weights stored in a single file, such as a ckpt or safetensors file. These single file types are typically produced from community trained models. There are three classes for loading single file weights:

  • [FromSingleFileMixin] supports loading pretrained pipeline weights stored in a single file, which can either be a ckpt or safetensors file.
  • [FromOriginalVAEMixin] supports loading a pretrained [AutoencoderKL] from pretrained ControlNet weights stored in a single file, which can either be a ckpt or safetensors file.
  • [FromOriginalControlnetMixin] supports loading pretrained ControlNet weights stored in a single file, which can either be a ckpt or safetensors file.

To learn more about how to load single file weights, see the Load different Stable Diffusion formats loading guide.

FromSingleFileMixin

autodoc loaders.single_file.FromSingleFileMixin

FromOriginalVAEMixin

autodoc loaders.single_file.FromOriginalVAEMixin

FromOriginalControlnetMixin

autodoc loaders.single_file.FromOriginalControlnetMixin