How to use EleutherAI/pythia-6.9b-squaring-first with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("EleutherAI/pythia-6.9b-squaring-first", device_map="auto")
