wheattoast11 commited on
Commit
7044a88
·
verified ·
1 Parent(s): e33c018

Upload eval_finetuned_v5.py with huggingface_hub

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