Improve benchmark scripts & add more models (#484)

2024-05-27 14:13:26 -07:00
parent 55c1643627
commit 09de730dee
6 changed files with 1173 additions and 11 deletions
--- a/benchmark/mmlu/bench_other.py
+++ b/benchmark/mmlu/bench_other.py
@@ -170,4 +170,4 @@ if __name__ == "__main__":
    parser.add_argument("--data_dir", type=str, default="data")
    parser.add_argument("--nsub", type=int, default=60)
    args = add_common_other_args_and_parse(parser)
-    main(args)
+    main(args)