1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00
Files
diffusers/docs/source/en/api/loaders/single_file.md
Sayak Paul 30e5e81d58 change to 2024 in the license (#6902)
change to 2024
2024-02-08 08:19:31 -10: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.autoencoder.FromOriginalVAEMixin

FromOriginalControlnetMixin

autodoc loaders.controlnet.FromOriginalControlNetMixin