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