update requirements and parse_args

uiuc-focal-lab · Oct 29, 2023 · 8c23f13 · 8c23f13
1 parent a19814a
commit 8c23f13
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/llm_cfg/infer.py b/llm_cfg/infer.py
@@ -26,7 +26,7 @@
     p.add_argument("--gpu", type=int, default=1)
     p.add_argument("--num_samples", type=int, default=1)
     p.add_argument("--language", choices = ["python", "go"], default = "python", help = "language")
-    p.add_argument("--dataset", choices = ["mbxp", "multi-humaneval", "mathqa-x"], default = "mbxp", help = "dataset")
+    p.add_argument("--dataset", choices = ["mbxp", "humaneval", "mathqa"], default = "humaneval", help = "dataset")
     args = p.parse_args()
 
     num_samples_per_task = args.num_samples

diff --git a/requirements.txt b/requirements.txt
@@ -41,3 +41,4 @@ transformers==4.28.1
 triton==2.0.0
 typing_extensions==4.7.1
 urllib3==2.0.4
+mxeval