diffusers
/

controlnet-canny-sdxl-1.0

stable-diffusion-xl

stable-diffusion-xl-diffusers

Model card Files Files and versions Community

TuringRM commited on Aug 10, 2023

Commit

eb51733

•

1 Parent(s): bdce74a

Set the VAE as the one downloaded

Files changed (1) hide show

README.md +1 -0

README.md CHANGED Viewed

@@ -63,6 +63,7 @@ vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype
 pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
     "stabilityai/stable-diffusion-xl-base-1.0",
     controlnet=controlnet,
     torch_dtype=torch.float16,
 )
 pipe.enable_model_cpu_offload()

 pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
     "stabilityai/stable-diffusion-xl-base-1.0",
     controlnet=controlnet,
+    vae=vae,
     torch_dtype=torch.float16,
 )
 pipe.enable_model_cpu_offload()