Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
ComfyUI
Commits
4656273e
Commit
4656273e
authored
Mar 09, 2024
by
MoonRide303
Browse files
Added additional nodes for CLIP merging
parent
a9ee9589
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
46 additions
and
0 deletions
+46
-0
comfy_extras/nodes_model_merging.py
comfy_extras/nodes_model_merging.py
+46
-0
No files found.
comfy_extras/nodes_model_merging.py
View file @
4656273e
...
...
@@ -87,6 +87,50 @@ class CLIPMergeSimple:
m
.
add_patches
({
k
:
kp
[
k
]},
1.0
-
ratio
,
ratio
)
return
(
m
,
)
class
CLIPSubtract
:
@
classmethod
def
INPUT_TYPES
(
s
):
return
{
"required"
:
{
"clip1"
:
(
"CLIP"
,),
"clip2"
:
(
"CLIP"
,),
"multiplier"
:
(
"FLOAT"
,
{
"default"
:
1.0
,
"min"
:
-
10.0
,
"max"
:
10.0
,
"step"
:
0.01
}),
}}
RETURN_TYPES
=
(
"CLIP"
,)
FUNCTION
=
"merge"
CATEGORY
=
"advanced/model_merging"
def
merge
(
self
,
clip1
,
clip2
,
multiplier
):
m
=
clip1
.
clone
()
kp
=
clip2
.
get_key_patches
()
for
k
in
kp
:
if
k
.
endswith
(
".position_ids"
)
or
k
.
endswith
(
".logit_scale"
):
continue
m
.
add_patches
({
k
:
kp
[
k
]},
-
multiplier
,
multiplier
)
return
(
m
,
)
class
CLIPAdd
:
@
classmethod
def
INPUT_TYPES
(
s
):
return
{
"required"
:
{
"clip1"
:
(
"CLIP"
,),
"clip2"
:
(
"CLIP"
,),
}}
RETURN_TYPES
=
(
"CLIP"
,)
FUNCTION
=
"merge"
CATEGORY
=
"advanced/model_merging"
def
merge
(
self
,
clip1
,
clip2
):
m
=
clip1
.
clone
()
kp
=
clip2
.
get_key_patches
()
for
k
in
kp
:
if
k
.
endswith
(
".position_ids"
)
or
k
.
endswith
(
".logit_scale"
):
continue
m
.
add_patches
({
k
:
kp
[
k
]},
1.0
,
1.0
)
return
(
m
,
)
class
ModelMergeBlocks
:
@
classmethod
def
INPUT_TYPES
(
s
):
...
...
@@ -279,6 +323,8 @@ NODE_CLASS_MAPPINGS = {
"ModelMergeAdd"
:
ModelAdd
,
"CheckpointSave"
:
CheckpointSave
,
"CLIPMergeSimple"
:
CLIPMergeSimple
,
"CLIPMergeSubtract"
:
CLIPSubtract
,
"CLIPMergeAdd"
:
CLIPAdd
,
"CLIPSave"
:
CLIPSave
,
"VAESave"
:
VAESave
,
}
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment