stabilityai

/

stable-diffusion-xl-base-1.0

like 6.23k

Follow

Stability AI 13.1k

base = DiffusionPipeline.from_pretrained(
"stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16, variant="fp16", use_safetensors=True
)

AttributeError Traceback (most recent call last)
in <cell line: 5>()
3
4 # load both base & refiner
----> 5 base = DiffusionPipeline.from_pretrained(
6 "stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16, variant="fp16", use_safetensors=True
7 )

/usr/local/lib/python3.10/dist-packages/diffusers/pipeline_utils.py in from_pretrained(cls, pretrained_model_name_or_path, **kwargs)
533 else:
534 diffusers_module = importlib.import_module(cls.module.split(".")[0])
--> 535 pipeline_class = getattr(diffusers_module, config_dict["_class_name"])
536
537 # To be removed in 1.0.0

AttributeError: module 'diffusers' has no attribute 'StableDiffusionXLPipeline'