diff --git a/ext/video_agent_deploy.py b/ext/video_agent_deploy.py index 08a009d..f5e8c44 100644 --- a/ext/video_agent_deploy.py +++ b/ext/video_agent_deploy.py @@ -58,7 +58,8 @@ vol = modal.Volume.from_name("comfy_model", environment_name="dev", create_if_mi ) @modal.web_server(8000, startup_timeout=120) def ui_1(): - subprocess.Popen("comfy launch -- --cpu --listen 0.0.0.0 --port 8000", shell=True) + process = subprocess.Popen("comfy launch -- --cpu --listen 0.0.0.0 --port 8000", shell=True) + process.wait() @app.function( diff --git a/nodes/img_agent.py b/nodes/img_agent.py index dcd5724..266bdfb 100644 --- a/nodes/img_agent.py +++ b/nodes/img_agent.py @@ -64,17 +64,10 @@ def fetch_and_process_image_models(): for model in data["data"]: tech_name = model.get("model_name") - if not tech_name: continue - - # --- 核心修正:不再手动添加任何前缀 --- - - # 1. 直接从API获取description,并用strip()清理首尾空格 + if not tech_name: + continue description_from_api = str(model.get("description", tech_name)).strip() - - # 2. 生成最终的显示名称 (description本身已包含前缀) display_name = f"{description_from_api} ({tech_name})" - - # 3. 仅根据mode分配排序键,用于内部排序 mode = model.get("mode") sort_key = 99 if mode == "t2i": diff --git a/nodes/union_llm_node.py b/nodes/union_llm_node.py index 7172f7b..9d27393 100644 --- a/nodes/union_llm_node.py +++ b/nodes/union_llm_node.py @@ -234,6 +234,7 @@ class LLMUionNode: return (result,) + # 节点映射 -# NODE_CLASS_MAPPINGS = {"LLMUionNode": LLMUionNode} -# NODE_DISPLAY_NAME_MAPPINGS = {"LLMUionNode": "聚合LLM节点(视频/图像/音频)"} +NODE_CLASS_MAPPINGS = {"LLMUionNode": LLMUionNode} +NODE_DISPLAY_NAME_MAPPINGS = {"LLMUionNode": "聚合LLM节点(视频/图像/音频)"}