Commit
•
56194ea
1
Parent(s):
bb3959c
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]>
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()
|