Update processing/two_stage/two_stage_processor.py

processing/two_stage/two_stage_processor.py

@@ -719,9 +719,9 @@ def _process_single_video(self,
     def cleanup(self):
         """Clean up temporary files and memory."""
         try:
-            # Clean up handlers first
-            self.memory_manager.cleanup_model(self.sam2_handler, "SAM2")
-            self.memory_manager.cleanup_model(self.matanyone_handler, "MatAnyone")
+            # Deep cleanup of handlers - we're done with processing
+            self.memory_manager.cleanup_model(self.sam2_handler, "SAM2", deep_cleanup=True)
+            self.memory_manager.cleanup_model(self.matanyone_handler, "MatAnyone", deep_cleanup=True)
             
             # Force complete memory cleanup
             self.memory_manager.cleanup_stage("Cleanup", force=True)