multimodalart HF staff TuringRM commited on
Commit
9e9060f
1 Parent(s): ea33f71

Set the VAE as the one downloaded (#6)

Browse files

- Set the VAE as the one downloaded (eb51733960875b6dd5b7ae31d5cf2e43b576d3dc)


Co-authored-by: Cesar Ramos Medina <[email protected]>

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()