Skip to content

Commit e324e19

Browse files
committed
resolve model name
1 parent 5e7f19d commit e324e19

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

lmms_eval/loggers/evaluation_tracker.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -68,7 +68,7 @@ def extract_model_name(model_args: str, key: str) -> str:
6868
return args_after_key.split(",")[0]
6969

7070
# order does matter, e.g. peft and delta are provided together with pretrained
71-
prefixes = ["peft=", "delta=", "pretrained=", "model=", "path=", "engine=", "model_version=", "model_name=", "model_id="]
71+
prefixes = ["peft=", "delta=", "pretrained=", "model=", "model_version=", "model_name=", "model_id=", "path=", "engine="]
7272
for prefix in prefixes:
7373
if prefix in model_args:
7474
return extract_model_name(model_args, prefix)

0 commit comments

Comments
 (0)