wheattoast11 commited on
Commit
b88042a
·
verified ·
1 Parent(s): a0bc111

Upload eval_finetuned_v4.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. eval_finetuned_v4.py +24 -0
eval_finetuned_v4.py ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ # "peft>=0.7.0",
10
+ # ]
11
+ # ///
12
+
13
+ """Evaluate fine-tuned wheattoast11/agent-zero-lfm-1.2b-v1."""
14
+
15
+ import subprocess
16
+ import sys
17
+
18
+ model_args = "model_name=wheattoast11/agent-zero-lfm-1.2b-v1,trust_remote_code=True"
19
+ tasks = "leaderboard|mmlu|5,leaderboard|arc:challenge|25,leaderboard|truthfulqa:mc|0"
20
+
21
+ cmd = ["lighteval", "vllm", model_args, tasks, "--output-dir", "./eval_finetuned"]
22
+ print(f"Running: {' '.join(cmd)}")
23
+ result = subprocess.run(cmd)
24
+ sys.exit(result.returncode)