298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-04-20 03:13:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

Now the model merge blocks node will use the longest match.

Browse Source
This commit is contained in:
comfyanonymous 2023-07-04 00:51:17 -04:00
parent 3a09fac835
commit c02f3baeaf
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy_extras/nodes_model_merging.py
View File

@ -46,9 +46,11 @@ class ModelMergeBlocks:
ratio = default_ratio ratio = default_ratio
k_unet = k[len("diffusion_model."):] k_unet = k[len("diffusion_model."):]
last_arg_size = 0
for arg in kwargs: for arg in kwargs:
if k_unet.startswith(arg): if k_unet.startswith(arg) and last_arg_size < len(arg):
ratio = kwargs[arg] ratio = kwargs[arg]
last_arg_size = len(arg)
m.add_patches({k: (sd[k], )}, 1.0 - ratio, ratio) m.add_patches({k: (sd[k], )}, 1.0 - ratio, ratio)
return (m, ) return (m, )