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