mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2025-06-09 07:37:14 +00:00
Slightly better fix for #7687
This commit is contained in:
parent
fd27494441
commit
2c735c13b4
@ -736,6 +736,7 @@ def load_controlnet_state_dict(state_dict, model=None, model_options={}):
|
|||||||
return control
|
return control
|
||||||
|
|
||||||
def load_controlnet(ckpt_path, model=None, model_options={}):
|
def load_controlnet(ckpt_path, model=None, model_options={}):
|
||||||
|
model_options = model_options.copy()
|
||||||
if "global_average_pooling" not in model_options:
|
if "global_average_pooling" not in model_options:
|
||||||
filename = os.path.splitext(ckpt_path)[0]
|
filename = os.path.splitext(ckpt_path)[0]
|
||||||
if filename.endswith("_shuffle") or filename.endswith("_shuffle_fp16"): #TODO: smarter way of enabling global_average_pooling
|
if filename.endswith("_shuffle") or filename.endswith("_shuffle_fp16"): #TODO: smarter way of enabling global_average_pooling
|
||||||
|
Loading…
x
Reference in New Issue
Block a user