mirror of
https://github.com/huggingface/diffusers.git
synced 2025-12-06 12:34:13 +08:00
1.1 KiB
1.1 KiB
AutoModel
The AutoModel is designed to make it easy to load a checkpoint without needing to know the specific model class. AutoModel automatically retrieves the correct model class from the checkpoint config.json file.
from diffusers import AutoModel, AutoPipelineForText2Image
unet = AutoModel.from_pretrained("stable-diffusion-v1-5/stable-diffusion-v1-5", subfolder="unet")
pipe = AutoPipelineForText2Image.from_pretrained("stable-diffusion-v1-5/stable-diffusion-v1-5", unet=unet)
AutoModel
autodoc AutoModel - all - from_pretrained