fix gen_kwargs arg reading (#1607)

0d920e82 · achervyakov · GitHub · f871646f · 0d920e82
Unverified Commit 0d920e82 authored Mar 20, 2024 by achervyakov Committed by GitHub Mar 19, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

lm_eval/__main__.py lm_eval/__main__.py +1 -1

No files found.
--- a/lm_eval/__main__.py
+++ b/lm_eval/__main__.py
@@ -182,7 +182,7 @@ def setup_parser() -> argparse.ArgumentParser:
    )
    parser.add_argument(
        "--gen_kwargs",
-        type=dict,
+        type=str,
        default=None,
        help=(
            "String arguments for model generation on greedy_until tasks,"