madebyollin

/

sdxl-vae-fp16-fix

like 519

Hello! Can this be used with SDXL-Turbo? For me it generates black images:

from diffusers import DiffusionPipeline, AutoencoderKL
import torch

vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)

pipe = DiffusionPipeline.from_pretrained("stabilityai/sdxl-turbo", vae=vae, use_safetensors=True, torch_dtype=torch.float16, variant="fp16").to("cuda")

prompt = "A cinematic shot of a baby racoon wearing an intricate italian priest robe."

image = pipe(prompt=prompt, num_inference_steps=1, guidance_scale=0.0).images[0]