update

binoculars/detector.py

@@ -85,14 +85,6 @@ class Binoculars(object):
             raise ValueError(f"Invalid mode: {mode}")
         
     def free_memory(self) -> None:
-        """Explicitly free GPU memory by moving models to CPU and deleting them"""
-        print("Freeing model memory...")
-        if hasattr(self, 'observer_model') and self.observer_model is not None:
-            self.observer_model = self.observer_model.to('cpu')
-            
-        if hasattr(self, 'performer_model') and self.performer_model is not None:
-            self.performer_model = self.performer_model.to('cpu')
-        
         if torch.cuda.is_available():
             torch.cuda.empty_cache()
             torch.cuda.synchronize()