guardiancc
commited on
Commit
•
1af481f
1
Parent(s):
17c4e15
Update app.py
Browse files
app.py
CHANGED
@@ -30,6 +30,7 @@ state_dict = load_state_dict(model_file)
|
|
30 |
model, _, _, _, _ = ControlNetModel_Union._load_pretrained_model(
|
31 |
controlnet_model, state_dict, model_file, "xinsir/controlnet-union-sdxl-1.0"
|
32 |
)
|
|
|
33 |
|
34 |
vae = AutoencoderKL.from_pretrained(
|
35 |
"madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16
|
|
|
30 |
model, _, _, _, _ = ControlNetModel_Union._load_pretrained_model(
|
31 |
controlnet_model, state_dict, model_file, "xinsir/controlnet-union-sdxl-1.0"
|
32 |
)
|
33 |
+
model.to(device="cpu", dtype=torch.float16)
|
34 |
|
35 |
vae = AutoencoderKL.from_pretrained(
|
36 |
"madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16
|