How to use NeelNanda/test-attn-only with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("NeelNanda/test-attn-only", dtype="auto")
