
python retrieval_head_detection.py  --model_path mistralai/Mistral-7B-Instruct-v0.3 --s 1000 --e 31500
python retrieval_head_detection.py  --model_path meta-llama/Llama-3.1-8B-Instruct --s 1000 --e 50000