save code

node_wrappers/pose_keypoint_postprocess.py

@@ -286,6 +286,9 @@ class RenderPeopleKpsMask:
                     "max": 1.0,      # 最大值
                     "step": 0.01     # 步进值
                 }),
+                "cloth_short": ("BOOLEAN", {
+                    "default": True
+                }),
             }
         }
 
@@ -293,7 +296,7 @@ class RenderPeopleKpsMask:
     FUNCTION = "render"
     CATEGORY = "ControlNet Preprocessors/Pose Keypoint Postprocess"
 
-    def render(self, kps, cloth_len, line_len) -> tuple[torch.Tensor]:
+    def render(self, kps, cloth_len, line_len, cloth_short) -> tuple[torch.Tensor]:
         if isinstance(kps, list):
             kps = kps[0]
 
@@ -304,6 +307,7 @@ class RenderPeopleKpsMask:
             width,
             cloth_len,
             line_len,
+            cloth_short,
         )
         return (mask,)