fruits=("coherence")
# fruits=("alpha" "alpha_hat" "stable_rank" "effective_rank" "ZD" "head_diversity" "coherence")
keywords=("self_attn" "mlp")
modles=("Llama-2-7b-hf" "Qwen2.5-7B")

for model in "${modles[@]}"; do
    for keyword in "${keywords[@]}"; do
        for fruit in "${fruits[@]}"; do
            echo "Begin main_assign: $model $keyword  $fruit"
            python -u main_assign.py --model_name $model --keyword $keyword  --metric_name $fruit
        done
    done
done