From 8e898fad4caab59bf4144e0cf11978b893de7e54 Mon Sep 17 00:00:00 2001
From: RandomGitUser321 <ruinerx@gmail.com>
Date: Wed, 6 Nov 2024 08:04:04 -0500
Subject: [PATCH] Add mochi to CLIPLoaderGGUF options (#154)

* Add mochi to CLIPLoaderGGUF options

ComfyUI implemented it today, so this should be all that's needed to use the T5 GGUF with it

* Update nodes.py
---
 nodes.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/nodes.py b/nodes.py
index 0f27df7..a8d61ba 100644
--- a/nodes.py
+++ b/nodes.py
@@ -287,6 +287,7 @@ clip_name_dict = {
     "sdxl": comfy.sd.CLIPType.STABLE_DIFFUSION,
     "sd3": comfy.sd.CLIPType.SD3,
     "flux": comfy.sd.CLIPType.FLUX,
+    "mochi": getattr(comfy.sd.CLIPType, "MOCHI", None),
 }
 
 class CLIPLoaderGGUF:
@@ -295,7 +296,7 @@ class CLIPLoaderGGUF:
         return {
             "required": {
                 "clip_name": (s.get_filename_list(),),
-                "type": (["stable_diffusion", "stable_cascade", "sd3", "stable_audio"],),
+                "type": (["stable_diffusion", "stable_cascade", "sd3", "stable_audio", "mochi"],),
             }
         }