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
3711b31d
Commit
3711b31d
authored
Feb 19, 2024
by
comfyanonymous
Browse files
Support Stable Cascade in checkpoint format.
parent
d91f45ef
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
6 additions
and
0 deletions
+6
-0
comfy/supported_models.py
comfy/supported_models.py
+6
-0
No files found.
comfy/supported_models.py
View file @
3711b31d
...
...
@@ -319,6 +319,10 @@ class Stable_Cascade_C(supported_models_base.BASE):
"shift"
:
2.0
,
}
vae_key_prefix
=
[
"vae."
]
text_encoder_key_prefix
=
[
"text_encoder."
]
clip_vision_prefix
=
"clip_l_vision."
def
process_unet_state_dict
(
self
,
state_dict
):
key_list
=
list
(
state_dict
.
keys
())
for
y
in
[
"weight"
,
"bias"
]:
...
...
@@ -355,6 +359,8 @@ class Stable_Cascade_B(Stable_Cascade_C):
"shift"
:
1.0
,
}
clip_vision_prefix
=
None
def
get_model
(
self
,
state_dict
,
prefix
=
""
,
device
=
None
):
out
=
model_base
.
StableCascade_B
(
self
,
device
=
device
)
return
out
...
...
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