https://colab.research.google.com/drive/1NCBOLPx038FxwEacmHDsCesWIAW1z8kU

Rewrite copy scores and dataset code to generalize beyond IOI. Change the template it takes in, and which tokens it looks for. Test it on ‘most recent subject’ important heads found by attention head analysis.