Fix single image segment offload

2025-09-28 11:35:08 +03:00
parent ebdb986959
commit 56abdb5a1e
1 changed files with 3 additions and 2 deletions
@@ -411,8 +411,9 @@ class Sam2Segmentation:
                model.to(offload_device)
            except:
                model.model.to(offload_device)
-            model.reset_state(self.inference_state)
-            self.inference_state = None
+            if hasattr(self, 'inference_state') and self.inference_state is not None and hasattr(model, "reset_state"):
+                model.reset_state(self.inference_state)
+                self.inference_state = None
            mm.soft_empty_cache()
        
        out_list = []