Update app.py
Browse files
app.py
CHANGED
|
@@ -29,13 +29,15 @@ model_file = hf_hub_download(
|
|
| 29 |
state_dict = load_state_dict(model_file)
|
| 30 |
# 获取 state_dict 中的所有键作为 loaded_keys
|
| 31 |
loaded_keys = list(state_dict.keys())
|
| 32 |
-
|
| 33 |
controlnet_model,
|
| 34 |
state_dict,
|
| 35 |
model_file,
|
| 36 |
"xinsir/controlnet-union-sdxl-1.0",
|
| 37 |
-
loaded_keys
|
| 38 |
)
|
|
|
|
|
|
|
| 39 |
model.to(device="cpu", dtype=torch.float16)
|
| 40 |
|
| 41 |
vae = AutoencoderKL.from_pretrained(
|
|
|
|
| 29 |
state_dict = load_state_dict(model_file)
|
| 30 |
# 获取 state_dict 中的所有键作为 loaded_keys
|
| 31 |
loaded_keys = list(state_dict.keys())
|
| 32 |
+
model_tuple = ControlNetModel_Union._load_pretrained_model(
|
| 33 |
controlnet_model,
|
| 34 |
state_dict,
|
| 35 |
model_file,
|
| 36 |
"xinsir/controlnet-union-sdxl-1.0",
|
| 37 |
+
loaded_keys
|
| 38 |
)
|
| 39 |
+
model = model_tuple[0] # 取出实际的模型对象
|
| 40 |
+
|
| 41 |
model.to(device="cpu", dtype=torch.float16)
|
| 42 |
|
| 43 |
vae = AutoencoderKL.from_pretrained(
|