Update README.md
Browse files
README.md
CHANGED
|
@@ -115,6 +115,7 @@ controlnet_conditioning_scale = 0.5
|
|
| 115 |
controlnet = ControlNetModel.from_pretrained(
|
| 116 |
"TheMistoAI/MistoLine",
|
| 117 |
torch_dtype=torch.float16
|
|
|
|
| 118 |
)
|
| 119 |
vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
|
| 120 |
pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
|
|
|
|
| 115 |
controlnet = ControlNetModel.from_pretrained(
|
| 116 |
"TheMistoAI/MistoLine",
|
| 117 |
torch_dtype=torch.float16
|
| 118 |
+
variant="fp16",
|
| 119 |
)
|
| 120 |
vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
|
| 121 |
pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
|