Update AILab_SegmentV2.py

2025-06-01 14:25:45 -07:00
parent 4b5163f56e
commit a3bc7460b3
1 changed files with 2 additions and 2 deletions
@@ -268,10 +268,10 @@ class SegmentV2:
        return (pil2tensor(result_image), mask_tensor, mask_image_vis)

 NODE_CLASS_MAPPINGS = {
-    "AILab_SegmentV2": SegmentV2,
+    "SegmentV2": SegmentV2,
 }

 NODE_DISPLAY_NAME_MAPPINGS = {
-    "AILab_SegmentV2": "Segmentation V2 (RMBG)",
+    "SegmentV2": "Segmentation V2 (RMBG)",
 }