save code

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,)