Set the VAE as the one downloaded
#6
by
TuringRM
- opened
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()
|