Make E5 adapter more restrict to reduce mismatch (lm-sys#2381)

wangzhen263 · Sep 7, 2023 · 56744d1 · 56744d1
1 parent 1d703b2
commit 56744d1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/fastchat/model/model_adapter.py b/fastchat/model/model_adapter.py
@@ -1405,7 +1405,7 @@ class E5Adapter(BaseModelAdapter):
     use_fast_tokenizer = False
 
     def match(self, model_path: str):
-        return "e5" in model_path.lower()
+        return "e5-" in model_path.lower()
 
     def load_model(self, model_path: str, from_pretrained_kwargs: dict):
         revision = from_pretrained_kwargs.get("revision", "main")