Upload 2 files

clip_analyzer.py

@@ -44,6 +44,12 @@ class CLIPAnalyzer:
             )
             self.tokenizer = open_clip.get_tokenizer(model_name)
             print(f"OpenCLIP model loaded successfully.")
+            import gc
+            gc.collect()
+            if torch.cuda.is_available():
+                torch.cuda.empty_cache()
+                torch.cuda.synchronize()
+            print("Memory cleanup completed after OpenCLIP loading.")
         except Exception as e:
             print(f"Error loading OpenCLIP model: {e}")
             raise

clip_model_manager.py

@@ -48,6 +48,15 @@ class CLIPModelManager:
             )
             self.tokenizer = open_clip.get_tokenizer(self.model_name)
             self.logger.info("Successfully loaded OpenCLIP model")
+            
+            # 立即清理 OpenCLIP 載入過程中的記憶體碎片
+            import gc
+            gc.collect()
+            if torch.cuda.is_available():
+                torch.cuda.empty_cache()
+                torch.cuda.synchronize()
+            self.logger.info("Memory cleanup completed after OpenCLIP loading in CLIPModelManager")
+            
         except Exception as e:
             self.logger.error(f"Error loading CLIP model: {e}")
             self.logger.error(traceback.format_exc())