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
55f37baa
Commit
55f37baa
authored
Mar 07, 2024
by
comfyanonymous
Browse files
Move some stable cascade nodes outside of _for_testing.
parent
3f75419e
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
3 deletions
+3
-3
comfy_extras/nodes_stable_cascade.py
comfy_extras/nodes_stable_cascade.py
+3
-3
No files found.
comfy_extras/nodes_stable_cascade.py
View file @
55f37baa
...
@@ -37,7 +37,7 @@ class StableCascade_EmptyLatentImage:
...
@@ -37,7 +37,7 @@ class StableCascade_EmptyLatentImage:
RETURN_NAMES
=
(
"stage_c"
,
"stage_b"
)
RETURN_NAMES
=
(
"stage_c"
,
"stage_b"
)
FUNCTION
=
"generate"
FUNCTION
=
"generate"
CATEGORY
=
"
_for_testing
/stable_cascade"
CATEGORY
=
"
latent
/stable_cascade"
def
generate
(
self
,
width
,
height
,
compression
,
batch_size
=
1
):
def
generate
(
self
,
width
,
height
,
compression
,
batch_size
=
1
):
c_latent
=
torch
.
zeros
([
batch_size
,
16
,
height
//
compression
,
width
//
compression
])
c_latent
=
torch
.
zeros
([
batch_size
,
16
,
height
//
compression
,
width
//
compression
])
...
@@ -63,7 +63,7 @@ class StableCascade_StageC_VAEEncode:
...
@@ -63,7 +63,7 @@ class StableCascade_StageC_VAEEncode:
RETURN_NAMES
=
(
"stage_c"
,
"stage_b"
)
RETURN_NAMES
=
(
"stage_c"
,
"stage_b"
)
FUNCTION
=
"generate"
FUNCTION
=
"generate"
CATEGORY
=
"
_for_testing
/stable_cascade"
CATEGORY
=
"
latent
/stable_cascade"
def
generate
(
self
,
image
,
vae
,
compression
):
def
generate
(
self
,
image
,
vae
,
compression
):
width
=
image
.
shape
[
-
2
]
width
=
image
.
shape
[
-
2
]
...
@@ -91,7 +91,7 @@ class StableCascade_StageB_Conditioning:
...
@@ -91,7 +91,7 @@ class StableCascade_StageB_Conditioning:
FUNCTION
=
"set_prior"
FUNCTION
=
"set_prior"
CATEGORY
=
"
_for_test
ing/stable_cascade"
CATEGORY
=
"
condition
ing/stable_cascade"
def
set_prior
(
self
,
conditioning
,
stage_c
):
def
set_prior
(
self
,
conditioning
,
stage_c
):
c
=
[]
c
=
[]
...
...
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