How to use EleutherAI/TinyStories-restricted with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("EleutherAI/TinyStories-restricted", dtype="auto")