jozee commited on
Commit
c28fb33
·
verified ·
1 Parent(s): 2aa3be8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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
- "madebbyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16
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(