Skip to content

Commit c199c8d

Browse files
authored
Fix Hermes2Adapter (#2727)
1 parent a754c48 commit c199c8d

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

fastchat/model/model_adapter.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1481,9 +1481,9 @@ class Hermes2Adapter(BaseModelAdapter):
14811481
use_fast_tokenizer = False
14821482

14831483
def match(self, model_path: str):
1484-
return (
1485-
"openhermes-2.5-mistral-7b"
1486-
or "openhermes-2-mistral-7b" in model_path.lower()
1484+
return any(
1485+
model_str in model_path.lower()
1486+
for model_str in ["openhermes-2.5-mistral-7b", "openhermes-2-mistral-7b"]
14871487
)
14881488

14891489
def load_model(self, model_path: str, from_pretrained_kwargs: dict):

0 commit comments

Comments
 (0)