Update models/model_manager.py
Browse files- 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 模型加载完成")
|