Set the VAE as the one downloaded

#6
by TuringRM - opened
Files changed (1) hide show
  1. README.md +1 -0
README.md CHANGED
@@ -63,6 +63,7 @@ vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype
63
  pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
64
  "stabilityai/stable-diffusion-xl-base-1.0",
65
  controlnet=controlnet,
 
66
  torch_dtype=torch.float16,
67
  )
68
  pipe.enable_model_cpu_offload()
 
63
  pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
64
  "stabilityai/stable-diffusion-xl-base-1.0",
65
  controlnet=controlnet,
66
+ vae=vae,
67
  torch_dtype=torch.float16,
68
  )
69
  pipe.enable_model_cpu_offload()