File size: 329 Bytes
e0d26dc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
# MiniRNN-256
---
license: cc
---
A small word-level RNN language model.
## Architecture
- Embedding size: 256
- RNN layers: 2
- Hidden size: 256
- Vocabulary size: 681
## Loading
```python
from safetensors.torch import load_file
state_dict = load_file("model.safetensors")
model.load_state_dict(state_dict)
model.eval()
``` |