Downloads · 30 days
27
6% of all-time downloads
mamba-370m-story-generation
DdIiVvYyAaMm/mamba-370m-story-generation
mamba-370m-story-generation is a text generation model from DdIiVvYyAaMm. Use it when you need the model to write or continue text. It is set up for transformers. The card lists the license as apache-2.0.
The State-Space/Mamba-370M is finetuned on ROC Stories dataset to be able to generate endings to short stories cohesively.
All-time downloads
483
Public
Parameters
159M
637 MB on disk
Likes
1
Public
Repo makeup
Click a slice to open those files.
.safetensors637 MB · 100%
At a glance
- Task
- Text Generation
- Library
- transformers
- License
- apache-2.0
- Model type
- mamba
- Access
- Public
- Created
- Apr 25, 2024
- Updated
- Apr 26, 2024
- SHA
- f6cde405
Try a prompt
Model card
From the Hugging Face model README
The State-Space/Mamba-370M is finetuned on ROC Stories dataset to be able to generate endings to short stories cohesively.
The Evaluation metrics on the ROC stories dataset for story ending generation are:
Bert (f1) : 0.878
Meteor: 0.1
bleu : 0.0125
Rouge1: 0.18
Perplexity : 207
To use the Model:
>>> from transformers import MambaForCausalLM, AutoTokenizer
>>> model_name = "DdIiVvYyAaMm/mamba-370m-story-generation"
>>> tokenizer = AutoTokenizer.from_pretrained(model_name)
>>> model = MambaForCausalLM.from_pretrained(model_name)
# And the rest of code standard as from transformers library.
- Task
- Text Generation
- Library
- transformers
- Type
- mamba
- License
- apache-2.0
- Languages
- en
- Created
- Apr 25, 2024
- Updated
- Apr 26, 2024