fix: SamAutoMaskSEGS error

2024-11-06 17:02:46 +08:00
parent 8b79c93fe5
commit 79a5fefc5d
1 changed files with 1 additions and 1 deletions
@@ -229,7 +229,7 @@ class SamAutoMaskSEGS(SamAutoMaskSEGSAdvanced):
    # OUTPUT_IS_LIST = (False, False)

    def generate(self, sam_model, image, output_mode):
-        masks = super().generate(sam_model, image, output_mode)
+        masks = super().generate(sam_model, image, output_mode=output_mode)
        masksRle = json.JSONEncoder().encode(masks)
        return {"ui": {"segsRle": (masksRle,)}, "result": (masksRle,)}