Bias Localization

This repository provides an implementation to localize bias in Llama-3.

Attention head files:

• attention_head_pruning.ipynb
• attn_main.py

Neuron files:

• neuron_pruning_main.py
• pipeline_pruning_clean.ipynb