Spaces:
Running
on
Zero
Running
on
Zero
A newer version of the Gradio SDK is available:
5.34.2
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