check if model attr of model exists

embeddings-benchmark · Nov 26, 2024 · b3304a3 · b3304a3
1 parent 3ff38ec
commit b3304a3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mteb/evaluation/evaluators/RetrievalEvaluator.py b/mteb/evaluation/evaluators/RetrievalEvaluator.py
@@ -318,7 +318,7 @@ def search_cross_encoder(
                 len(queries_in_pair) == len(corpus_in_pair) == len(instructions_in_pair)
             )
 
-            if isinstance(self.model.model, CrossEncoder):
+            if hasattr(self.model, "model") and isinstance(self.model.model, CrossEncoder):
                 # can't take instructions, so add them here
                 queries_in_pair = [
                     f"{q} {i}".strip()
@@ -428,7 +428,7 @@ def encode(
 
 
 def is_cross_encoder_compatible(model) -> bool:
-    op = getattr(model.model, "predict", None)
+    op = getattr(model, "predict", None)
     return callable(op)