File size: 362 Bytes
a38f44b |
1 2 3 4 5 6 7 8 |
#!/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)
|