wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-25 13:50:15 +08:00
Code Issues Actions 15 Packages Projects Releases Wiki Activity

Support "lite" version of alibaba-pai Z-Image Controlnet

Browse Source 
* reduced number of control layers (3) compared to full model
This commit is contained in:
Acly 2026-01-13 17:17:47 +01:00
parent 5ac1372533
commit 71780085a7
No known key found for this signature in database
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy_extras/nodes_model_patch.py
View File

@ -244,6 +244,10 @@ class ModelPatchLoader:
elif 'control_all_x_embedder.2-1.weight' in sd: # alipai z image fun controlnet elif 'control_all_x_embedder.2-1.weight' in sd: # alipai z image fun controlnet
sd = z_image_convert(sd) sd = z_image_convert(sd)
config = {} config = {}
if 'control_layers.4.adaLN_modulation.0.weight' not in sd:
config['n_control_layers'] = 3
config['additional_in_dim'] = 17
config['refiner_control'] = True
if 'control_layers.14.adaLN_modulation.0.weight' in sd: if 'control_layers.14.adaLN_modulation.0.weight' in sd:
config['n_control_layers'] = 15 config['n_control_layers'] = 15
config['additional_in_dim'] = 17 config['additional_in_dim'] = 17