Open more model nodes

Author: longgui0318

.style.yapf

@@ -0,0 +1,5 @@
+[style]
+based_on_style = pep8
+column_limit = 230
+SPLIT_BEFORE_FIRST_ARGUMENT = true
+SPLIT_ARGUMENTS_WHEN_COMMA_TERMINATED = true

README.md

@@ -20,42 +20,134 @@ Before using this node, you need to have a WaveSpeed AI API key. You can obtain
 
 5. Start ComfyUI and enjoy using the WaveSpeed AI API node!
 
-## Nodes
 
-### Client
+## How to Use
 
-This node is used to create a WaveSpeed AI client.
-
-### Wan Text to Video (T2V)
-
-This node generates a video from a text prompt using the Wan2.1 T2V model.
-
-<p align="center">
-  <img src="./examples/wan-t2v-lora.png" alt="Wan T2V with LoRA">
-</p>
-
-Example workflow: [wan-t2v-lora.json](./examples/wan-t2v-lora.json)
-
-Sample output: [wavespeed_video_1743346194.mp4](./examples/wavespeed_video_1743346194.mp4)
-
-### Flux Image to Image (I2I)
-
-This node generates images from input images using the Flux Dev model with LoRA support.
-
-<p align="center">
-  <img src="./examples/flux-dev-lora-i2i.png" alt="Flux I2I with LoRA">
-</p>
-
-Example workflow: [flux-dev-lora-i2i.json](./examples/flux-dev-lora-i2i.json)
-
-### Flux Text to Image (T2I)
-
-This node generates images from text prompts using the Flux Dev model with LoRA support.
-
-<p align="center">
-  <img src="./examples/flux-dev-lora-t2i.png" alt="Flux T2I with LoRA">
-</p>
-
-### Preview Video
-
-This node is used to preview and save videos generated by the WaveSpeed AI API.
+The following are typical workflows and result demonstrations (each group includes a ComfyUI workflow screenshot).
+The workflow images contain workflow information and can be directly dragged into ComfyUI for use.
+
+---
+
+#### 1. Dia TTS
+- Workflow Example:
+
+  ![Dia TTS Workflow](examples/dia_tts.png)
+
+---
+
+#### 2. Flux Control LoRA Canny
+- Workflow Example:
+
+  ![Flux Control LoRA Canny Workflow](examples/flux_control_lora_canny.png)
+
+---
+
+#### 3. Flux Dev Lora Ultra Fast 
+- Workflow Example:
+
+  ![Flux Dev Lora Ultra Fast Workflow](examples/flux_dev_lora_ultra_fast.png)
+
+---
+
+#### 4. Hunyuan Custom Ref2V 720p Workflow and Result
+- Workflow Example:
+
+  ![Hunyuan Custom Ref2V 720p Workflow](examples/hunyuan_custom_ref2v_720p.png)
+
+- Result Video:
+
+  <video src="examples/hunyuan_custom_ref2v_720p.mp4" controls width="480"></video>
+
+---
+
+#### 5. Wan2.1 I2V 720p Ultra Fast Workflow and Result
+- Workflow Example:
+
+  ![Wan2.1 I2V 720p Ultra Fast Workflow](examples/wan_2_1_i2v_720p_ultra_fast.png)
+
+- Result Video:
+
+  <video src="examples/wan_2_1_i2v_720p_ultra_fast.mp4" controls width="480"></video>
+
+---
+
+
+### Here are some popular models you can experience in ComfyUI:
+1. Hunyuan Custom
+2. Ghibli
+3. Wan2.1
+
+### Here are some nodes list
+* "WaveSpeedAI Client"
+* "WaveSpeedAI Dia TTS"
+* "WaveSpeedAI Flux Control LoRA Canny"
+* "WaveSpeedAI Flux Control LoRA Depth"
+* "WaveSpeedAI Flux Dev"
+* "WaveSpeedAI Flux Dev Fill"
+* "WaveSpeedAI Flux Dev Lora"
+* "WaveSpeedAI Flux Dev Lora Ultra Fast"
+* "WaveSpeedAI Flux Dev Ultra Fast"
+* "WaveSpeedAI Flux Pro Redux"
+* "WaveSpeedAI Flux Redux Dev"
+* "WaveSpeedAI Flux Schnell"
+* "WaveSpeedAI Flux Schnell Lora"
+* "WaveSpeedAI Flux and SDXL Loras"
+* "WaveSpeedAI Framepack"
+* "WaveSpeedAI Ghibli"
+* "WaveSpeedAI Hidream E1 Full"
+* "WaveSpeedAI Hidream I1 Dev"
+* "WaveSpeedAI Hidream I1 Full"
+* "WaveSpeedAI Hunyuan 3D V2 Multi View"
+* "WaveSpeedAI Hunyuan Custom Ref2V 480p"
+* "WaveSpeedAI Hunyuan Custom Ref2V 720p"
+* "WaveSpeedAI Hunyuan Video I2V"
+* "WaveSpeedAI Hunyuan Video T2V"
+* "WaveSpeedAI Instant Character"
+* "WaveSpeedAI Kling v1.6 I2V Pro"
+* "WaveSpeedAI Kling v1.6 I2V Standard"
+* "WaveSpeedAI Kling v1.6 T2V Standard"
+* "WaveSpeedAI LTX Video I2V 480p"
+* "WaveSpeedAI LTX Video I2V 720p"
+* "WaveSpeedAI MMAudio V2"
+* "WaveSpeedAI Magi 1.24b"
+* "WaveSpeedAI Minimax Video 01"
+* "WaveSpeedAI Preview Video"
+* "WaveSpeedAI Real-ESRGAN"
+* "WaveSpeedAI SDXL"
+* "WaveSpeedAI SDXL Lora"
+* "WaveSpeedAI Save Audio"
+* "WaveSpeedAI SkyReels V1"
+* "WaveSpeedAI Step1X Edit"
+* "WaveSpeedAI Uno"
+* "WaveSpeedAI Upload Image"
+* "WaveSpeedAI Vidu Image to Video2.0"
+* "WaveSpeedAI Vidu Reference To Video2.0"
+* "WaveSpeedAI Vidu Start/End To Video2.0"
+* "WaveSpeedAI Wan Loras"
+* "WaveSpeedAI Wan2.1 I2V 480p"
+* "WaveSpeedAI Wan2.1 I2V 480p LoRA Ultra Fast"
+* "WaveSpeedAI Wan2.1 I2V 480p Lora"
+* "WaveSpeedAI Wan2.1 I2V 480p Ultra Fast"
+* "WaveSpeedAI Wan2.1 I2V 720p"
+* "WaveSpeedAI Wan2.1 I2V 720p LoRA Ultra Fast"
+* "WaveSpeedAI Wan2.1 I2V 720p Lora"
+* "WaveSpeedAI Wan2.1 I2V 720p Ultra Fast"
+* "WaveSpeedAI Wan2.1 T2V 480p LoRA"
+* "WaveSpeedAI Wan2.1 T2V 480p LoRA Ultra Fast"
+* "WaveSpeedAI Wan2.1 T2V 480p Ultra Fast"
+* "WaveSpeedAI Wan2.1 T2V 720p"
+* "WaveSpeedAI Wan2.1 T2V 720p LoRA"
+* "WaveSpeedAI Wan2.1 T2V 720p LoRA Ultra Fast"
+* "WaveSpeedAI Wan2.1 T2V 720p Ultra Fast"
+
+### How to Apply Lora
+1. As we provide services on WaveSpeedAI-API, you cannot use your local lora files. However, we support loading lora via URL.
+2. You can use "WaveSpeedAi Wan Loras", "WaveSpeedAi Flux Loras", or "WaveSpeedAi Flux SDXL Loras" nodes.
+3. Enter the lora URL in the lora_path field. For example: https://huggingface.co/WaveSpeedAi/WanLoras/resolve/main/wan_loras.safetensors
+4. Enter the lora weight in the lora_weight field. For example: 0.5
+5. If you have multiple loras, you can add additional lora_path and lora_weight pairs.
+6. If your model is not on Hugging Face, that's fine. Any publicly accessible URL will work.
+
+### How to Use image_url in Nodes
+1. You can use the "WaveSpeedAi Upload Image" node to convert a local IMAGE into an image_url.
+2. Connect the output to the corresponding node that requires it. You can find examples in the provided samples.

__init__.py

@@ -1,4 +1,55 @@
-from .py import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
+import importlib.util
+import os
+import sys
+
+NODE_CLASS_MAPPINGS = {}
+NODE_DISPLAY_NAME_MAPPINGS = {}
+
+python = sys.executable
+
+def get_ext_dir(subpath=None, mkdir=False):
+    dir = os.path.dirname(__file__)
+    if subpath is not None:
+        dir = os.path.join(dir, subpath)
+
+    dir = os.path.abspath(dir)
+
+    if mkdir and not os.path.exists(dir):
+        os.makedirs(dir)
+    return dir
+
+def serialize(obj):
+    if isinstance(obj, (str, int, float, bool, list, dict, type(None))):
+        return obj
+    return str(obj)  # 转为字符串
+
+
+py = get_ext_dir("py")
+files = os.listdir(py)
+
+all_nodes = {}
+for file in files:
+    if not file.endswith(".py"):
+        continue
+    name = os.path.splitext(file)[0]
+    imported_module = importlib.import_module(".py.{}".format(name), __name__)
+    try:
+        NODE_CLASS_MAPPINGS = {**NODE_CLASS_MAPPINGS, **imported_module.NODE_CLASS_MAPPINGS}
+        NODE_DISPLAY_NAME_MAPPINGS = {**NODE_DISPLAY_NAME_MAPPINGS, **imported_module.NODE_DISPLAY_NAME_MAPPINGS}
+        serialized_CLASS_MAPPINGS = {k: serialize(v) for k, v in imported_module.NODE_CLASS_MAPPINGS.items()}
+        serialized_DISPLAY_NAME_MAPPINGS = {k: serialize(v) for k, v in imported_module.NODE_DISPLAY_NAME_MAPPINGS.items()}
+        all_nodes[file]={"NODE_CLASS_MAPPINGS": serialized_CLASS_MAPPINGS, "NODE_DISPLAY_NAME_MAPPINGS": serialized_DISPLAY_NAME_MAPPINGS}
+    except:
+        pass
+print("------------------------")
+names_list = []
+for key, value in NODE_DISPLAY_NAME_MAPPINGS.items():
+    names_list.append(value)
+names_list.sort()
+
+for name in names_list:
+    print(f"\"{name}\"")
+print("------------------------")
 
 WEB_DIRECTORY = "./web"