Spaces:

MogensR
/

VideoBackgroundReplacer

Paused

MogensR commited on Aug 23

Commit

34c8dd9

1 Parent(s): 0d886c8

Update model_loader.py

Files changed (1) hide show

model_loader.py CHANGED Viewed

@@ -367,7 +367,7 @@ def _cleanup_models(self):
             self.matanyone_core = None
         # Clear GPU cache
-        self.memory_manager.cleanup_gpu_memory()
         gc.collect()
         logger.debug("Model cleanup completed")

             self.matanyone_core = None
         # Clear GPU cache
+        self.memory_manager.cleanup_aggressive()
         gc.collect()
         logger.debug("Model cleanup completed")