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