wheattoast11 commited on
Commit
e33c018
·
verified ·
1 Parent(s): 3828ada

Upload eval_baseline_v5.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. eval_baseline_v5.py +22 -0
eval_baseline_v5.py ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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",
8
+ # ]
9
+ # ///
10
+
11
+ """Evaluate baseline LiquidAI/LFM2.5-1.2B-Instruct using accelerate backend."""
12
+
13
+ import subprocess
14
+ import sys
15
+
16
+ model_args = "model_name=LiquidAI/LFM2.5-1.2B-Instruct,trust_remote_code=True,dtype=bfloat16"
17
+ tasks = "leaderboard|mmlu|5,leaderboard|arc:challenge|25,leaderboard|truthfulqa:mc|0"
18
+
19
+ cmd = ["lighteval", "accelerate", model_args, tasks, "--output-dir", "./eval_baseline"]
20
+ print(f"Running: {' '.join(cmd)}")
21
+ result = subprocess.run(cmd)
22
+ sys.exit(result.returncode)