File size: 689 Bytes
67d4be3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
# /// script
# requires-python = ">=3.10"
# dependencies = [
# "lighteval[vllm]>=0.6.0",
# "torch>=2.0.0",
# "transformers>=4.40.0",
# "accelerate>=0.30.0",
# "vllm>=0.4.0",
# "peft>=0.7.0",
# ]
# ///
"""Evaluate fine-tuned wheattoast11/agent-zero-lfm-1.2b-v1 with lighteval vllm."""
import subprocess
import sys
model = "wheattoast11/agent-zero-lfm-1.2b-v1"
tasks = "leaderboard|mmlu|5,leaderboard|arc:challenge|25,leaderboard|truthfulqa:mc|0"
cmd = ["lighteval", "vllm", model, tasks, "--trust-remote-code", "--use-chat-template", "--output-dir", "./eval_finetuned"]
print(f"Running: {' '.join(cmd)}")
result = subprocess.run(cmd)
sys.exit(result.returncode)
|