Update models/model_manager.py

models/model_manager.py

@@ -197,10 +197,20 @@ class ModelManager:
         color = (random.randint(120, 200), random.randint(120, 200), random.randint(120, 200))
         return Image.new('RGB', (width, height), color=color)
 
-    def cleanup(self):
-        logger.info("释放模型占用显存和缓存...")
-        try:
-            
-            # ...清理代码不变...
-        except Exception as e:
-            logger.error(f"清理显存失败: {e}")
+def cleanup(self):
+    logger.info("释放模型占用显存和缓存...")
+    try:
+        # 清理模型引用
+        del self.caption_model
+        del self.clip_model
+        del self.sd_pipeline
+        del self.controlnet
+        del self.controlnet_pipeline
+        
+        # 清理缓存
+        if torch.cuda.is_available():
+            torch.cuda.empty_cache()
+        
+        logger.info("显存和缓存清理完成")
+    except Exception as e:
+        logger.error(f"清理显存失败: {e}")