tools / bug_5776.py
patrickvonplaten's picture
[LCM] Better error message
a38f44b
raw
history blame
362 Bytes
#!/usr/bin/env python3
from diffusers import AutoPipelineForText2Image
import torch
pipeline = AutoPipelineForText2Image.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16).to("cuda")
pipeline.load_lora_weights("ostris/super-cereal-sdxl-lora", weight_name="cereal_box_sdxl_v1.safetensors")
print(pipeline.unet.attn_processor)