StableCascade CLIP model support.

2024-02-16 13:29:04 -05:00
parent 667c92814e
commit 97d03ae04a
5 changed files with 43 additions and 8 deletions
--- a/comfy/supported_models.py
+++ b/comfy/supported_models.py
@@ -336,7 +336,7 @@ class Stable_Cascade_C(supported_models_base.BASE):
        return out

    def clip_target(self):
-        return None
+        return supported_models_base.ClipTarget(sdxl_clip.StableCascadeTokenizer, sdxl_clip.StableCascadeClipModel)

 class Stable_Cascade_B(Stable_Cascade_C):
    unet_config = {