wheattoast11 commited on
Commit
59141be
·
verified ·
1 Parent(s): 4c6bb3c

Upload eval_lfm_finetuned.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. eval_lfm_finetuned.py +5 -3
eval_lfm_finetuned.py CHANGED
@@ -1,7 +1,7 @@
1
  # /// script
2
  # requires-python = ">=3.10"
3
  # dependencies = [
4
- # "lighteval[accelerate]>=0.6.0",
5
  # "torch>=2.0.0",
6
  # "transformers>=4.40.0",
7
  # "accelerate>=0.30.0",
@@ -14,12 +14,14 @@
14
  import subprocess
15
  import sys
16
 
 
17
  tasks = "leaderboard|mmlu|5,leaderboard|hellaswag|0,leaderboard|arc_challenge|25"
18
 
 
19
  cmd = [
20
  sys.executable, "-m", "lighteval", "accelerate",
21
- "--model-args", "pretrained=wheattoast11/agent-zero-lfm-1.2b-v1,trust_remote_code=True",
22
- "--tasks", tasks,
23
  "--output-dir", "./eval_results_finetuned",
24
  ]
25
 
 
1
  # /// script
2
  # requires-python = ">=3.10"
3
  # dependencies = [
4
+ # "lighteval>=0.6.0",
5
  # "torch>=2.0.0",
6
  # "transformers>=4.40.0",
7
  # "accelerate>=0.30.0",
 
14
  import subprocess
15
  import sys
16
 
17
+ model_args = "pretrained=wheattoast11/agent-zero-lfm-1.2b-v1,trust_remote_code=True"
18
  tasks = "leaderboard|mmlu|5,leaderboard|hellaswag|0,leaderboard|arc_challenge|25"
19
 
20
+ # lighteval 0.13+ uses positional args: accelerate MODEL_ARGS TASKS
21
  cmd = [
22
  sys.executable, "-m", "lighteval", "accelerate",
23
+ model_args,
24
+ tasks,
25
  "--output-dir", "./eval_results_finetuned",
26
  ]
27