mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-02-01 17:20:17 +08:00
bbbdc8430b
Add search aliases to model-related and miscellaneous nodes: - Model nodes: nodes_model_merging.py, nodes_model_advanced.py, nodes_lora_extract.py - Sampler nodes: nodes_custom_sampler.py, nodes_align_your_steps.py - Control nodes: nodes_controlnet.py, nodes_attention_multiply.py, nodes_hooks.py - Training nodes: nodes_train.py, nodes_dataset.py - Utility nodes: nodes_logic.py, nodes_canny.py, nodes_differential_diffusion.py - Architecture-specific: nodes_sd3.py, nodes_pixart.py, nodes_lumina2.py, nodes_kandinsky5.py, nodes_hidream.py, nodes_fresca.py, nodes_hunyuan3d.py - Media nodes: nodes_load_3d.py, nodes_webcam.py, nodes_preview_any.py, nodes_wanmove.py Uses search_aliases parameter in io.Schema() for v3 nodes, SEARCH_ALIASES class attribute for legacy nodes.
75 lines
2.9 KiB
Python
75 lines
2.9 KiB
Python
from typing_extensions import override
|
|
|
|
import folder_paths
|
|
import comfy.sd
|
|
import comfy.model_management
|
|
from comfy_api.latest import ComfyExtension, io
|
|
|
|
|
|
class QuadrupleCLIPLoader(io.ComfyNode):
|
|
@classmethod
|
|
def define_schema(cls):
|
|
return io.Schema(
|
|
node_id="QuadrupleCLIPLoader",
|
|
category="advanced/loaders",
|
|
description="[Recipes]\n\nhidream: long clip-l, long clip-g, t5xxl, llama_8b_3.1_instruct",
|
|
inputs=[
|
|
io.Combo.Input("clip_name1", options=folder_paths.get_filename_list("text_encoders")),
|
|
io.Combo.Input("clip_name2", options=folder_paths.get_filename_list("text_encoders")),
|
|
io.Combo.Input("clip_name3", options=folder_paths.get_filename_list("text_encoders")),
|
|
io.Combo.Input("clip_name4", options=folder_paths.get_filename_list("text_encoders")),
|
|
],
|
|
outputs=[
|
|
io.Clip.Output(),
|
|
]
|
|
)
|
|
|
|
@classmethod
|
|
def execute(cls, clip_name1, clip_name2, clip_name3, clip_name4):
|
|
clip_path1 = folder_paths.get_full_path_or_raise("text_encoders", clip_name1)
|
|
clip_path2 = folder_paths.get_full_path_or_raise("text_encoders", clip_name2)
|
|
clip_path3 = folder_paths.get_full_path_or_raise("text_encoders", clip_name3)
|
|
clip_path4 = folder_paths.get_full_path_or_raise("text_encoders", clip_name4)
|
|
clip = comfy.sd.load_clip(ckpt_paths=[clip_path1, clip_path2, clip_path3, clip_path4], embedding_directory=folder_paths.get_folder_paths("embeddings"))
|
|
return io.NodeOutput(clip)
|
|
|
|
class CLIPTextEncodeHiDream(io.ComfyNode):
|
|
@classmethod
|
|
def define_schema(cls):
|
|
return io.Schema(
|
|
node_id="CLIPTextEncodeHiDream",
|
|
search_aliases=["hidream prompt"],
|
|
category="advanced/conditioning",
|
|
inputs=[
|
|
io.Clip.Input("clip"),
|
|
io.String.Input("clip_l", multiline=True, dynamic_prompts=True),
|
|
io.String.Input("clip_g", multiline=True, dynamic_prompts=True),
|
|
io.String.Input("t5xxl", multiline=True, dynamic_prompts=True),
|
|
io.String.Input("llama", multiline=True, dynamic_prompts=True),
|
|
],
|
|
outputs=[
|
|
io.Conditioning.Output(),
|
|
]
|
|
)
|
|
|
|
@classmethod
|
|
def execute(cls, clip, clip_l, clip_g, t5xxl, llama):
|
|
tokens = clip.tokenize(clip_g)
|
|
tokens["l"] = clip.tokenize(clip_l)["l"]
|
|
tokens["t5xxl"] = clip.tokenize(t5xxl)["t5xxl"]
|
|
tokens["llama"] = clip.tokenize(llama)["llama"]
|
|
return io.NodeOutput(clip.encode_from_tokens_scheduled(tokens))
|
|
|
|
|
|
class HiDreamExtension(ComfyExtension):
|
|
@override
|
|
async def get_node_list(self) -> list[type[io.ComfyNode]]:
|
|
return [
|
|
QuadrupleCLIPLoader,
|
|
CLIPTextEncodeHiDream,
|
|
]
|
|
|
|
|
|
async def comfy_entrypoint() -> HiDreamExtension:
|
|
return HiDreamExtension()
|