Humphreykowl commited on
Commit
c06b67e
·
verified ·
1 Parent(s): 4aeeadd

Update models/model_manager.py

Browse files
Files changed (1) hide show
  1. models/model_manager.py +2 -1
models/model_manager.py CHANGED
@@ -88,7 +88,8 @@ class ModelManager:
88
 
89
  # 启用内存优化
90
  if hasattr(self.caption_model, 'enable_attention_slicing'):
91
- # # self.caption_model.enable_attention_slicing() # Removed: BLIP does not support attention slicing # Removed: BLIP does not support attention slicing
 
92
 
93
  self.caption_model.eval()
94
  logger.info("BLIP 模型加载完成")
 
88
 
89
  # 启用内存优化
90
  if hasattr(self.caption_model, 'enable_attention_slicing'):
91
+ pass # BLIP 不支持 attention slicing
92
+
93
 
94
  self.caption_model.eval()
95
  logger.info("BLIP 模型加载完成")