|
|
|
pip install antlr4-python3-runtime==4.11 immutabledict langdetect nltk lm_eval |
|
|
|
|
|
python -c "import nltk; nltk.download('punkt')" |
|
|
|
MODEL_PATHS=( |
|
huihui-ai/Llama-3.2-3B-Instruct-abliterated |
|
) |
|
|
|
for MODEL_PATH in "${MODEL_PATHS[@]}"; do |
|
MODEL_NAME=$(basename "$MODEL_PATH") |
|
MODEL_DIR="./results/$MODEL_NAME" |
|
mkdir -p "$MODEL_DIR" |
|
|
|
MODEL_ARGS="trust_remote_code=True,pretrained=$MODEL_PATH,dtype=bfloat16" |
|
|
|
BASE_COMMAND="accelerate launch -m lm_eval --model hf --model_args $MODEL_ARGS --batch_size 4 --fewshot_as_multiturn --apply_chat_template" |
|
|
|
|
|
$BASE_COMMAND --tasks leaderboard_ifeval --fewshot_as_multiturn --output_path "$MODEL_DIR/ifeval" |
|
|
|
|
|
$BASE_COMMAND --tasks leaderboard_bbh --num_fewshot 3 --fewshot_as_multiturn --output_path "$MODEL_DIR/bbh" |
|
|
|
|
|
$BASE_COMMAND --tasks leaderboard_gpqa --fewshot_as_multiturn --output_path "$MODEL_DIR/gpqa" |
|
|
|
|
|
$BASE_COMMAND --tasks leaderboard_mmlu_pro --num_fewshot 5 --fewshot_as_multiturn --output_path "$MODEL_DIR/mmlu_pro" |
|
|
|
|
|
$BASE_COMMAND --tasks truthfulqa_mc2 --fewshot_as_multiturn --output_path "$MODEL_DIR/truthfulqa" |
|
|
|
|
|
done |
|
|