Spaces:
Runtime error
Runtime error
Update models/model_manager.py
Browse files- models/model_manager.py +0 -2
models/model_manager.py
CHANGED
|
@@ -87,8 +87,6 @@ class ModelManager:
|
|
| 87 |
).to(self.device)
|
| 88 |
|
| 89 |
# 启用内存优化
|
| 90 |
-
if hasattr(self.caption_model, 'enable_attention_slicing'):
|
| 91 |
-
pass # BLIP 不支持 attention slicing
|
| 92 |
|
| 93 |
|
| 94 |
self.caption_model.eval()
|
|
|
|
| 87 |
).to(self.device)
|
| 88 |
|
| 89 |
# 启用内存优化
|
|
|
|
|
|
|
| 90 |
|
| 91 |
|
| 92 |
self.caption_model.eval()
|