From 9e421e5b99178cb56f2a8c68485fb2061d0d3732 Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Wed, 26 Jun 2024 08:38:42 -0400 Subject: [PATCH] update openpose-xl controlnet --- modules/control/units/controlnet.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/modules/control/units/controlnet.py b/modules/control/units/controlnet.py index b57005f3c..138c5fba9 100644 --- a/modules/control/units/controlnet.py +++ b/modules/control/units/controlnet.py @@ -49,7 +49,8 @@ predefined_sdxl = { 'Canny XL': 'diffusers/controlnet-canny-sdxl-1.0', 'Depth Zoe XL': 'diffusers/controlnet-zoe-depth-sdxl-1.0', 'Depth Mid XL': 'diffusers/controlnet-depth-sdxl-1.0-mid', - 'OpenPose XL': 'thibaud/controlnet-openpose-sdxl-1.0', + 'OpenPose XL': 'thibaud/controlnet-openpose-sdxl-1.0/bin', + # 'OpenPose XL': 'thibaud/controlnet-openpose-sdxl-1.0/OpenPoseXL2.safetensors', 'Xinsir OpenPose XL': 'xinsir/controlnet-openpose-sdxl-1.0', 'Xinsir Canny XL': 'xinsir/controlnet-canny-sdxl-1.0', 'Xinsir Scribble XL': 'xinsir/controlnet-scribble-sdxl-1.0', @@ -171,6 +172,9 @@ class ControlNet(): if model_path.endswith('.safetensors'): self.load_safetensors(model_path) else: + if '/bin' in model_path: + model_path = model_path.replace('/bin', '') + self.load_config['use_safetensors'] = False self.model = ControlNetModel.from_pretrained(model_path, **self.load_config) if self.dtype is not None: self.model.to(self.dtype)