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