| #!/usr/bin/env python3 | |
| #@title Fuse/unfuse LoRAs sequentially leading to trouble | |
| import torch | |
| from diffusers import StableDiffusionXLPipeline | |
| pipe = StableDiffusionXLPipeline.from_pretrained( | |
| "stabilityai/stable-diffusion-xl-base-1.0", | |
| torch_dtype=torch.float16, | |
| variant="fp16", | |
| ).to("cuda") | |
| pipe.load_lora_weights("Pclanglais/TintinIA") | |
| pipe.fuse_lora() | |
| images = pipe("a mecha robot", num_inference_steps=2) | |
| pipe.unfuse_lora() | |
| pipe.unload_lora_weights() | |
| pipe.load_lora_weights("ProomptEngineer/pe-balloon-diffusion-style") | |
| pipe.fuse_lora() | |
| images = pipe("a mecha robot", num_inference_steps=2) | |
| pipe.unfuse_lora() | |
| pipe.unload_lora_weights() | |
| pipe.load_lora_weights("ostris/crayon_style_lora_sdxl") | |
| pipe.fuse_lora() | |
| images = pipe("a mecha robot", num_inference_steps=2) | |
| pipe.unfuse_lora() | |
| pipe.unload_lora_weights() | |
| pipe.load_lora_weights("joachimsallstrom/aether-cloud-lora-for-sdxl") | |
| pipe.fuse_lora() | |
| images = pipe("a mecha robot", num_inference_steps=2) | |
| pipe.unfuse_lora() | |
| pipe.unload_lora_weights() | |