kevinwang676's picture
Upload folder using huggingface_hub
6755a2d verified
|
raw
history blame
1.78 kB

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