From 563ecd177411c1b72779cca0c94ff370d55161e7 Mon Sep 17 00:00:00 2001
From: colomi <1421901449@qq.com>
Date: Sat, 12 Jul 2025 19:35:18 +0800
Subject: [PATCH] save code

---
 node_wrappers/pose_keypoint_postprocess.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/node_wrappers/pose_keypoint_postprocess.py b/node_wrappers/pose_keypoint_postprocess.py
index 9baf21d..7bc6184 100644
--- a/node_wrappers/pose_keypoint_postprocess.py
+++ b/node_wrappers/pose_keypoint_postprocess.py
@@ -276,7 +276,10 @@ class RenderPeopleKpsMask:
         return {
             "required": {
                 "kps": ("POSE_KEYPOINT",),
-                "changdu": (["yao", "kuan", "da_tui", "xi_gai", "xiao_tui", "jiao_huai"], {"default": "kuan"}),
+                "cloth_len": ("STRING", {
+                    "multiline": False,  # 是否允许多行输入
+                    "default": "yao"  # 默认值
+                }),
             }
         }
 
@@ -284,7 +287,7 @@ class RenderPeopleKpsMask:
     FUNCTION = "render"
     CATEGORY = "ControlNet Preprocessors/Pose Keypoint Postprocess"
 
-    def render(self, kps, changdu) -> tuple[torch.Tensor]:
+    def render(self, kps, cloth_len) -> tuple[torch.Tensor]:
         if isinstance(kps, list):
             kps = kps[0]
 
@@ -293,7 +296,7 @@ class RenderPeopleKpsMask:
             poses,
             height,
             width,
-            changdu,
+            cloth_len,
         )
         return (mask,)