File size: 389 Bytes
11343d3 97d1293 11343d3 97d1293 11343d3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
---
license: cc
language:
- en
pipeline_tag: text-generation
tags:
- medical
---
# MiniRNN-256
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()
``` |