Skip to content

Commit

Permalink
fix
Browse files Browse the repository at this point in the history
  • Loading branch information
thomwolf committed Feb 7, 2024
1 parent 39befd2 commit 33dbd59
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/lighteval/main_nanotron.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down

0 comments on commit 33dbd59

Please sign in to comment.