MiniRNN-256

A small word-level RNN language model.

Architecture

• Embedding size: 256
• RNN layers: 2
• Hidden size: 256
• Vocabulary size: 681

Loading

from safetensors.torch import load_file

state_dict = load_file("model.safetensors")
model.load_state_dict(state_dict)
model.eval()