metadata
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
from safetensors.torch import load_file
state_dict = load_file("model.safetensors")
model.load_state_dict(state_dict)
model.eval()