test.sh

# bash head_base.sh \
#             7 \
#             ReasonKV \
#             128 \
#             flash_attention_2 \
#             meta-llama/Meta-Llama-3-8B-Instruct \
#             reason \
#             1.5 \
#             1

        # bash needle_base.sh \
        #     0 \
        #     ReasonKV \
        #     128 \
        #     flash_attention_2 \
        #     meta-llama/Meta-Llama-3-8B-Instruct \
        #     copy \
        #     1.5 \
        #     1 \
        #     1000 \
        #     8001 \
        #     1000 \
        #     LLaMA3 \
        #     Meta-Llama-3-8B-Instruct 


        bash head_base_babi.sh \
            0 \
            ReasonKV \
            128 \
            flash_attention_2 \
            meta-llama/Meta-Llama-3-8B-Instruct \
            reason \
            1.5 \
            1