Spaces:
Sleeping
Sleeping
fix(reranker): resolve AttributeError by fixing reranker initialization order
Browse files- models/reranker.py +1 -1
models/reranker.py
CHANGED
|
@@ -17,7 +17,6 @@ class TextReranker:
|
|
| 17 |
def __init__(self, model_name: str = "BAAI/bge-reranker-v2-m3", use_fp16: bool = False):
|
| 18 |
self.model_name = model_name
|
| 19 |
self.device = self._get_device()
|
| 20 |
-
self._warmup()
|
| 21 |
|
| 22 |
try:
|
| 23 |
logger.info(f"⏳ Loading Reranker Model: {self.model_name} on {self.device}")
|
|
@@ -25,6 +24,7 @@ class TextReranker:
|
|
| 25 |
self.model_name,
|
| 26 |
use_fp16=(use_fp16 and self.device.startswith("cuda"))
|
| 27 |
)
|
|
|
|
| 28 |
logger.info("✅ Reranker Model loaded successfully.")
|
| 29 |
except Exception as e:
|
| 30 |
logger.critical(f"❌ Failed to load Reranker Model: {e}", exc_info=True)
|
|
|
|
| 17 |
def __init__(self, model_name: str = "BAAI/bge-reranker-v2-m3", use_fp16: bool = False):
|
| 18 |
self.model_name = model_name
|
| 19 |
self.device = self._get_device()
|
|
|
|
| 20 |
|
| 21 |
try:
|
| 22 |
logger.info(f"⏳ Loading Reranker Model: {self.model_name} on {self.device}")
|
|
|
|
| 24 |
self.model_name,
|
| 25 |
use_fp16=(use_fp16 and self.device.startswith("cuda"))
|
| 26 |
)
|
| 27 |
+
self._warmup()
|
| 28 |
logger.info("✅ Reranker Model loaded successfully.")
|
| 29 |
except Exception as e:
|
| 30 |
logger.critical(f"❌ Failed to load Reranker Model: {e}", exc_info=True)
|