| | --- |
| | license: mit |
| | pipeline_tag: text-to-image |
| | library_name: diffusers |
| | --- |
| | # Diffusers API of Transparent Image Layer Diffusion using Latent Transparency |
| | Create transparent image with Diffusers! |
| |  |
| | Please check the Github repo [here](https://github.com/rootonchair/diffuser_layerdiffuse) |
| |
|
| | This is a port to Diffuser from original [SD Webui's Layer Diffusion](https://github.com/layerdiffusion/sd-forge-layerdiffuse) to extend the ability to generate transparent image with your favorite API |
| |
|
| |
|
| | Paper: [Transparent Image Layer Diffusion using Latent Transparency](https://arxiv.org/abs/2402.17113) |
| | ## Quickstart |
| |
|
| | Generate transparent image with SD1.5 models. In this example, we will use [digiplay/Juggernaut_final](https://huggingface.co/digiplay/Juggernaut_final) as the base model |
| |
|
| | ```python |
| | from huggingface_hub import hf_hub_download |
| | from safetensors.torch import load_file |
| | import torch |
| | |
| | from diffusers import StableDiffusionPipeline |
| | |
| | from models import TransparentVAEDecoder |
| | from loaders import load_lora_to_unet |
| | |
| | |
| | |
| | if __name__ == "__main__": |
| | |
| | model_path = hf_hub_download( |
| | 'LayerDiffusion/layerdiffusion-v1', |
| | 'layer_sd15_vae_transparent_decoder.safetensors', |
| | ) |
| | |
| | vae_transparent_decoder = TransparentVAEDecoder.from_pretrained("digiplay/Juggernaut_final", subfolder="vae", torch_dtype=torch.float16).to("cuda") |
| | vae_transparent_decoder.set_transparent_decoder(load_file(model_path)) |
| | |
| | pipeline = StableDiffusionPipeline.from_pretrained("digiplay/Juggernaut_final", vae=vae_transparent_decoder, torch_dtype=torch.float16, safety_checker=None).to("cuda") |
| | |
| | model_path = hf_hub_download( |
| | 'LayerDiffusion/layerdiffusion-v1', |
| | 'layer_sd15_transparent_attn.safetensors' |
| | ) |
| | |
| | load_lora_to_unet(pipeline.unet, model_path, frames=1) |
| | |
| | image = pipeline(prompt="a dog sitting in room, high quality", |
| | width=512, height=512, |
| | num_images_per_prompt=1, return_dict=False)[0] |
| | ``` |