From 33dbd5953dcaad6596519d363fcb5e574f5b1eb5 Mon Sep 17 00:00:00 2001 From: Thomas Wolf Date: Wed, 7 Feb 2024 11:15:38 +0000 Subject: [PATCH] fix --- src/lighteval/main_nanotron.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/lighteval/main_nanotron.py b/src/lighteval/main_nanotron.py index 9597f2734..a4e29e01d 100644 --- a/src/lighteval/main_nanotron.py +++ b/src/lighteval/main_nanotron.py @@ -54,7 +54,7 @@ def main( raise ValueError("The checkpoint path should point to a YAML file") nanotron_config: config_cls = get_config_from_file( - local_config_path, config_class=config_cls, model_config_class=model_config_cls, strict=False + local_config_path, config_class=config_cls, model_config_class=model_config_cls ) if lighteval_config_path: