Transformers How to use empty-michael/tinystories_1layer_attn_mlp_C10k_k16 with Transformers:
# Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("empty-michael/tinystories_1layer_attn_mlp_C10k_k16", device_map="auto") 
