🍰 Stable Diffusion (XL) 微型自动编码器

TAESDXL 是一款非常小巧的自动编码器，它使用与 SDXL-VAE 相同的“ latent API”。 TAESDXL 可用于实时预览 SDXL 生成过程。

在我的笔记本电脑上的对比：

image/jpeg

本仓库包含 TAESDXL 权重的 .safetensors 版本。

对于 SD1.x / SD2.x，请改用 TAESD（SD 和 SDXL 的 VAE 不兼容，相关讨论参见 https://huggingface.co/madebyollin/sdxl-vae-fp16-fix/discussions/6#64b8a9c13707b7d603c6ac16）。

在 🧨 diffusers 中使用

import torch
from diffusers import DiffusionPipeline, AutoencoderTiny


base = 'PyTorch-NPU/stable-diffusion-xl-base-1_0'
pipe = DiffusionPipeline.from_pretrained(
    base, torch_dtype=torch.float16
)

repo = 'OpenSource/taesdxl'
pipe.vae = AutoencoderTiny.from_pretrained(repo, torch_dtype=torch.float16)
pipe = pipe.to("npu:0")

prompt = "slice of delicious New York-style berry cheesecake"
image = pipe(prompt, num_inference_steps=25).images[0]
image.save("cheesecake_sdxl.png")

🍰 Stable Diffusion (XL) 微型自动编码器

TAESDXL 是一款非常小巧的自动编码器，它使用与 SDXL-VAE 相同的“ latent API”。 TAESDXL 可用于实时预览 SDXL 生成过程。

在我的笔记本电脑上的对比：

image/jpeg

本仓库包含 TAESDXL 权重的 .safetensors 版本。

对于 SD1.x / SD2.x，请改用 TAESD（SD 和 SDXL 的 VAE 不兼容，相关讨论参见 https://huggingface.co/madebyollin/sdxl-vae-fp16-fix/discussions/6#64b8a9c13707b7d603c6ac16）。

在 🧨 diffusers 中使用

import torch
from diffusers import DiffusionPipeline, AutoencoderTiny


base = 'PyTorch-NPU/stable-diffusion-xl-base-1_0'
pipe = DiffusionPipeline.from_pretrained(
    base, torch_dtype=torch.float16
)

repo = 'OpenSource/taesdxl'
pipe.vae = AutoencoderTiny.from_pretrained(repo, torch_dtype=torch.float16)
pipe = pipe.to("npu:0")

prompt = "slice of delicious New York-style berry cheesecake"
image = pipe(prompt, num_inference_steps=25).images[0]
image.save("cheesecake_sdxl.png")