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

Update models/model_manager.py

Browse files
Files changed (1) hide show
  1. models/model_manager.py +1 -1
models/model_manager.py CHANGED
@@ -88,7 +88,7 @@ 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
92
 
93
  self.caption_model.eval()
94
  logger.info("BLIP 模型加载完成")
 
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 模型加载完成")