From 186723f34d7bd19a36bcb3dd748bf816f21be64b Mon Sep 17 00:00:00 2001
From: John Pollock <pollockjj@gmail.com>
Date: Wed, 18 Dec 2024 17:01:01 -0600
Subject: [PATCH] Add mochi and ltxv support to CLIPLoaderMultiGPU

---
 __init__.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/__init__.py b/__init__.py
index 8a3ee41..f7e8c90 100644
--- a/__init__.py
+++ b/__init__.py
@@ -203,7 +203,7 @@ class CLIPLoaderMultiGPU:
             "required": {
                 "clip_name": (folder_paths.get_filename_list("clip"),),
                 "type": (
-                    ["stable_diffusion", "stable_cascade", "sd3", "stable_audio"],
+                    ["stable_diffusion", "stable_cascade", "sd3", "stable_audio", "mochi", "ltxv"],
                 ),
                 "device": ([f"cuda:{i}" for i in range(torch.cuda.device_count())],),
             }
@@ -223,6 +223,10 @@ class CLIPLoaderMultiGPU:
             clip_type = comfy.sd.CLIPType.SD3
         elif type == "stable_audio":
             clip_type = comfy.sd.CLIPType.STABLE_AUDIO
+        elif type == "mochi":
+            clip_type = comfy.sd.CLIPType.MOCHI
+        elif type == "ltxv":
+            clip_type = comfy.sd.CLIPType.LTXV
         else:
             clip_type = comfy.sd.CLIPType.STABLE_DIFFUSION