save code

node_wrappers/pose_keypoint_postprocess.py

@@ -280,6 +280,12 @@ class RenderPeopleKpsMask:
                     "multiline": False,  # 是否允许多行输入
                     "default": "yao"  # 默认值
                 }),
+                "line_len": ("FLOAT", {
+                    "default": 0.5,  # 默认值
+                    "min": 0.1,     # 最小值
+                    "max": 1.0,      # 最大值
+                    "step": 0.01     # 步进值
+                }),
             }
         }
 
@@ -287,7 +293,7 @@ class RenderPeopleKpsMask:
     FUNCTION = "render"
     CATEGORY = "ControlNet Preprocessors/Pose Keypoint Postprocess"
 
-    def render(self, kps, cloth_len) -> tuple[torch.Tensor]:
+    def render(self, kps, cloth_len, line_len) -> tuple[torch.Tensor]:
         if isinstance(kps, list):
             kps = kps[0]
 
@@ -297,6 +303,7 @@ class RenderPeopleKpsMask:
             height,
             width,
             cloth_len,
+            line_len,
         )
         return (mask,)