298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-01-25 15:55:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Support Stable Cascade in checkpoint format.

Browse Source
This commit is contained in:
comfyanonymous 2024-02-19 11:20:48 -05:00
parent d91f45ef28
commit 3711b31dff
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
comfy/supported_models.py
View File

@ -319,6 +319,10 @@ class Stable_Cascade_C(supported_models_base.BASE):
"shift": 2.0, "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): def process_unet_state_dict(self, state_dict):
key_list = list(state_dict.keys()) key_list = list(state_dict.keys())
for y in ["weight", "bias"]: for y in ["weight", "bias"]:
@ -355,6 +359,8 @@ class Stable_Cascade_B(Stable_Cascade_C):
"shift": 1.0, "shift": 1.0,
} }
clip_vision_prefix = None
def get_model(self, state_dict, prefix="", device=None): def get_model(self, state_dict, prefix="", device=None):
out = model_base.StableCascade_B(self, device=device) out = model_base.StableCascade_B(self, device=device)
return out return out