m97j commited on
Commit
1487b7f
·
1 Parent(s): 02a8a1c

fix(reranker): resolve AttributeError by fixing reranker initialization order

Browse files
Files changed (1) hide show
  1. 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)