Update app.py
Browse files
app.py
CHANGED
|
@@ -28,7 +28,11 @@ model_file = hf_hub_download(
|
|
| 28 |
)
|
| 29 |
state_dict = load_state_dict(model_file)
|
| 30 |
model = ControlNetModel_Union._load_pretrained_model(
|
| 31 |
-
controlnet_model,
|
|
|
|
|
|
|
|
|
|
|
|
|
| 32 |
)
|
| 33 |
model.to(device="cpu", dtype=torch.float16)
|
| 34 |
|
|
|
|
| 28 |
)
|
| 29 |
state_dict = load_state_dict(model_file)
|
| 30 |
model = ControlNetModel_Union._load_pretrained_model(
|
| 31 |
+
controlnet_model,
|
| 32 |
+
state_dict,
|
| 33 |
+
model_file,
|
| 34 |
+
"xinsir/controlnet-union-sdxl-1.0",
|
| 35 |
+
loaded_keys # 提供缺失的参数
|
| 36 |
)
|
| 37 |
model.to(device="cpu", dtype=torch.float16)
|
| 38 |
|