How to use taufeeque/TinyStories-1Layer-21M-Codebook with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("taufeeque/TinyStories-1Layer-21M-Codebook", dtype="auto")