MogensR commited on
Commit
7ec2f7a
·
1 Parent(s): db85143

agent 2.2

Browse files
Files changed (1) hide show
  1. models/matanyone_loader.py +9 -0
models/matanyone_loader.py CHANGED
@@ -989,6 +989,15 @@ def process_stream(
989
  log.error(error_msg, exc_info=True)
990
  _emit_progress(progress_cb, -1, f"CLEANUP ERROR: {error_msg}")
991
  raise MatAnyError(error_msg) from e
 
 
 
 
 
 
 
 
 
992
 
993
  def _flush_chunk(self, frames_bgr, seed_1hw, alpha_writer, fg_writer):
994
  """
 
989
  log.error(error_msg, exc_info=True)
990
  _emit_progress(progress_cb, -1, f"CLEANUP ERROR: {error_msg}")
991
  raise MatAnyError(error_msg) from e
992
+ finally:
993
+ # Ensure all resources are cleaned up
994
+ if 'cap' in locals() and cap is not None:
995
+ cap.release()
996
+ if 'alpha_writer' in locals() and alpha_writer is not None:
997
+ alpha_writer.release()
998
+ if 'fg_writer' in locals() and fg_writer is not None:
999
+ fg_writer.release()
1000
+ _safe_empty_cache()
1001
 
1002
  def _flush_chunk(self, frames_bgr, seed_1hw, alpha_writer, fg_writer):
1003
  """