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
9f71e4b6
Commit
9f71e4b6
authored
Mar 02, 2024
by
comfyanonymous
Browse files
Let model patches patch sub objects.
parent
00425563
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
3 deletions
+3
-3
comfy/model_patcher.py
comfy/model_patcher.py
+3
-3
No files found.
comfy/model_patcher.py
View file @
9f71e4b6
...
@@ -176,10 +176,10 @@ class ModelPatcher:
...
@@ -176,10 +176,10 @@ class ModelPatcher:
def
patch_model
(
self
,
device_to
=
None
,
patch_weights
=
True
):
def
patch_model
(
self
,
device_to
=
None
,
patch_weights
=
True
):
for
k
in
self
.
object_patches
:
for
k
in
self
.
object_patches
:
old
=
getattr
(
self
.
model
,
k
)
old
=
comfy
.
utils
.
get
_
attr
(
self
.
model
,
k
)
if
k
not
in
self
.
object_patches_backup
:
if
k
not
in
self
.
object_patches_backup
:
self
.
object_patches_backup
[
k
]
=
old
self
.
object_patches_backup
[
k
]
=
old
setattr
(
self
.
model
,
k
,
self
.
object_patches
[
k
])
comfy
.
utils
.
set
_
attr
(
self
.
model
,
k
,
self
.
object_patches
[
k
])
if
patch_weights
:
if
patch_weights
:
model_sd
=
self
.
model_state_dict
()
model_sd
=
self
.
model_state_dict
()
...
@@ -352,6 +352,6 @@ class ModelPatcher:
...
@@ -352,6 +352,6 @@ class ModelPatcher:
keys
=
list
(
self
.
object_patches_backup
.
keys
())
keys
=
list
(
self
.
object_patches_backup
.
keys
())
for
k
in
keys
:
for
k
in
keys
:
setattr
(
self
.
model
,
k
,
self
.
object_patches_backup
[
k
])
comfy
.
utils
.
set
_
attr
(
self
.
model
,
k
,
self
.
object_patches_backup
[
k
])
self
.
object_patches_backup
=
{}
self
.
object_patches_backup
=
{}
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