Update AILab_SAM3Segment.py

2026-01-06 01:24:40 -08:00
parent 7178f940fb
commit 112fd6141d
1 changed files with 2 additions and 1 deletions
@@ -304,7 +304,7 @@ class SAM3Segment:

        if unload_model:
            device_str = "cuda" if torch_device.type == "cuda" else "cpu"
-            cache_key = (sam3_model, device_str)
+            cache_key = ("sam3", device_str)
            if cache_key in self.processor_cache:
                del self.processor_cache[cache_key]
            if torch_device.type == "cuda":
@@ -330,3 +330,4 @@ class SAM3Segment:

 NODE_CLASS_MAPPINGS = {"SAM3Segment": SAM3Segment}
 NODE_DISPLAY_NAME_MAPPINGS = {"SAM3Segment": "SAM3 Segmentation (RMBG)"}
+