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 <TuringRM@users.noreply.huggingface.co>
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()
|