| 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() | |
| ``` |