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
ff6ca2a8
Commit
ff6ca2a8
authored
Jul 18, 2024
by
comfyanonymous
Browse files
Move PAG to model_patches/unet section.
Move other unet model_patches nodes to model_patches/unet section.
parent
374e093e
Changes
3
Show whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
4 additions
and
4 deletions
+4
-4
comfy_extras/nodes_freelunch.py
comfy_extras/nodes_freelunch.py
+2
-2
comfy_extras/nodes_hypertile.py
comfy_extras/nodes_hypertile.py
+1
-1
comfy_extras/nodes_pag.py
comfy_extras/nodes_pag.py
+1
-1
No files found.
comfy_extras/nodes_freelunch.py
View file @
ff6ca2a8
...
@@ -34,7 +34,7 @@ class FreeU:
...
@@ -34,7 +34,7 @@ class FreeU:
RETURN_TYPES
=
(
"MODEL"
,)
RETURN_TYPES
=
(
"MODEL"
,)
FUNCTION
=
"patch"
FUNCTION
=
"patch"
CATEGORY
=
"model_patches"
CATEGORY
=
"model_patches
/unet
"
def
patch
(
self
,
model
,
b1
,
b2
,
s1
,
s2
):
def
patch
(
self
,
model
,
b1
,
b2
,
s1
,
s2
):
model_channels
=
model
.
model
.
model_config
.
unet_config
[
"model_channels"
]
model_channels
=
model
.
model
.
model_config
.
unet_config
[
"model_channels"
]
...
@@ -73,7 +73,7 @@ class FreeU_V2:
...
@@ -73,7 +73,7 @@ class FreeU_V2:
RETURN_TYPES
=
(
"MODEL"
,)
RETURN_TYPES
=
(
"MODEL"
,)
FUNCTION
=
"patch"
FUNCTION
=
"patch"
CATEGORY
=
"model_patches"
CATEGORY
=
"model_patches
/unet
"
def
patch
(
self
,
model
,
b1
,
b2
,
s1
,
s2
):
def
patch
(
self
,
model
,
b1
,
b2
,
s1
,
s2
):
model_channels
=
model
.
model
.
model_config
.
unet_config
[
"model_channels"
]
model_channels
=
model
.
model
.
model_config
.
unet_config
[
"model_channels"
]
...
...
comfy_extras/nodes_hypertile.py
View file @
ff6ca2a8
...
@@ -32,7 +32,7 @@ class HyperTile:
...
@@ -32,7 +32,7 @@ class HyperTile:
RETURN_TYPES
=
(
"MODEL"
,)
RETURN_TYPES
=
(
"MODEL"
,)
FUNCTION
=
"patch"
FUNCTION
=
"patch"
CATEGORY
=
"model_patches"
CATEGORY
=
"model_patches
/unet
"
def
patch
(
self
,
model
,
tile_size
,
swap_size
,
max_depth
,
scale_depth
):
def
patch
(
self
,
model
,
tile_size
,
swap_size
,
max_depth
,
scale_depth
):
model_channels
=
model
.
model
.
model_config
.
unet_config
[
"model_channels"
]
model_channels
=
model
.
model
.
model_config
.
unet_config
[
"model_channels"
]
...
...
comfy_extras/nodes_pag.py
View file @
ff6ca2a8
...
@@ -19,7 +19,7 @@ class PerturbedAttentionGuidance:
...
@@ -19,7 +19,7 @@ class PerturbedAttentionGuidance:
RETURN_TYPES
=
(
"MODEL"
,)
RETURN_TYPES
=
(
"MODEL"
,)
FUNCTION
=
"patch"
FUNCTION
=
"patch"
CATEGORY
=
"
_for_testing
"
CATEGORY
=
"
model_patches/unet
"
def
patch
(
self
,
model
,
scale
):
def
patch
(
self
,
model
,
scale
):
unet_block
=
"middle"
unet_block
=
"middle"
...
...
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