wheattoast11 commited on
Commit
a6f934e
·
verified ·
1 Parent(s): e60aeeb

Upload eval_baseline_v3.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. eval_baseline_v3.py +23 -0
eval_baseline_v3.py ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # /// script
2
+ # requires-python = ">=3.10"
3
+ # dependencies = [
4
+ # "lighteval[vllm]>=0.6.0",
5
+ # "torch>=2.0.0",
6
+ # "transformers>=4.40.0",
7
+ # "accelerate>=0.30.0",
8
+ # "vllm>=0.4.0",
9
+ # ]
10
+ # ///
11
+
12
+ """Evaluate baseline LiquidAI/LFM2.5-1.2B-Instruct."""
13
+
14
+ import subprocess
15
+ import sys
16
+
17
+ model_args = "pretrained=LiquidAI/LFM2.5-1.2B-Instruct,trust_remote_code=True,add_bos_token=True"
18
+ tasks = "leaderboard|mmlu|5,leaderboard|arc:challenge|25,leaderboard|truthfulqa:mc|0"
19
+
20
+ cmd = ["lighteval", "vllm", model_args, tasks, "--output-dir", "./eval_baseline"]
21
+ print(f"Running: {' '.join(cmd)}")
22
+ result = subprocess.run(cmd)
23
+ sys.exit(result.returncode)