Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -30,7 +30,7 @@ model, _, _, _, _ = ControlNetModel_Union._load_pretrained_model(
|
|
30 |
model.to(device="cuda", dtype=torch.float16)
|
31 |
|
32 |
vae = AutoencoderKL.from_pretrained(
|
33 |
-
"
|
34 |
).to("cuda")
|
35 |
|
36 |
pipe = StableDiffusionXLFillPipeline.from_pretrained(
|
|
|
30 |
model.to(device="cuda", dtype=torch.float16)
|
31 |
|
32 |
vae = AutoencoderKL.from_pretrained(
|
33 |
+
"madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16
|
34 |
).to("cuda")
|
35 |
|
36 |
pipe = StableDiffusionXLFillPipeline.from_pretrained(
|