Skip to content

Commit a128918

Browse files
Add LTXAVTextEncoderLoader node.
1 parent f2b0023 commit a128918

File tree

1 file changed

+33
-0
lines changed

1 file changed

+33
-0
lines changed

comfy_extras/nodes_lt_audio.py

Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -169,13 +169,46 @@ def execute(
169169
)
170170

171171

172+
class LTXAVTextEncoderLoader(io.ComfyNode):
173+
@classmethod
174+
def define_schema(cls) -> io.Schema:
175+
return io.Schema(
176+
node_id="LTXAVTextEncoderLoader",
177+
display_name="LTXV Audio Text Encoder Loader",
178+
category="advanced/loaders",
179+
description="[Recipes]\n\nltxav: gemma 3 12B",
180+
inputs=[
181+
io.Combo.Input(
182+
"text_encoder",
183+
options=folder_paths.get_filename_list("text_encoders"),
184+
),
185+
io.Combo.Input(
186+
"ckpt_name",
187+
options=folder_paths.get_filename_list("checkpoints"),
188+
)
189+
],
190+
outputs=[io.Clip.Output(display_name="Audio VAE")],
191+
)
192+
193+
@classmethod
194+
def execute(cls, text_encoder, ckpt_name, device="default"):
195+
clip_type = comfy.sd.CLIPType.LTXV
196+
197+
clip_path1 = folder_paths.get_full_path_or_raise("text_encoders", text_encoder)
198+
clip_path2 = folder_paths.get_full_path_or_raise("checkpoints", ckpt_name)
199+
200+
clip = comfy.sd.load_clip(ckpt_paths=[clip_path1, clip_path2], embedding_directory=folder_paths.get_folder_paths("embeddings"), clip_type=clip_type)
201+
return io.NodeOutput(clip)
202+
203+
172204
class LTXVAudioExtension(ComfyExtension):
173205
async def get_node_list(self) -> list[type[io.ComfyNode]]:
174206
return [
175207
LTXVAudioVAELoader,
176208
LTXVAudioVAEEncode,
177209
LTXVAudioVAEDecode,
178210
LTXVEmptyLatentAudio,
211+
LTXAVTextEncoderLoader,
179212
]
180213

181214

0 commit comments

Comments
 (0)