wheattoast11 commited on
Commit
4c6bb3c
·
verified ·
1 Parent(s): 9c86ea0

Upload eval_lfm_baseline.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. eval_lfm_baseline.py +5 -3
eval_lfm_baseline.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",
@@ -13,12 +13,14 @@
13
  import subprocess
14
  import sys
15
 
 
16
  tasks = "leaderboard|mmlu|5,leaderboard|hellaswag|0,leaderboard|arc_challenge|25"
17
 
 
18
  cmd = [
19
  sys.executable, "-m", "lighteval", "accelerate",
20
- "--model-args", "pretrained=LiquidAI/LFM2.5-1.2B-Instruct,trust_remote_code=True",
21
- "--tasks", tasks,
22
  "--output-dir", "./eval_results_baseline",
23
  ]
24
 
 
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",
 
13
  import subprocess
14
  import sys
15
 
16
+ model_args = "pretrained=LiquidAI/LFM2.5-1.2B-Instruct,trust_remote_code=True"
17
  tasks = "leaderboard|mmlu|5,leaderboard|hellaswag|0,leaderboard|arc_challenge|25"
18
 
19
+ # lighteval 0.13+ uses positional args: accelerate MODEL_ARGS TASKS
20
  cmd = [
21
  sys.executable, "-m", "lighteval", "accelerate",
22
+ model_args,
23
+ tasks,
24
  "--output-dir", "./eval_results_baseline",
25
  ]
26