File size: 1,257 Bytes
aad5ead 2c94baf da4cd3f 2c94baf aad5ead 2c94baf |
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 27 28 29 30 31 32 33 34 35 36 37 38 39 40 |
---
license: mit
language:
- en
tags:
- NoSleep
- Reddit
- Story
- Horror
widget:
- text: "[WP] \""
example_title: "[WP] "
datasets:
- chloeliu/reddit_nosleep_posts
---
# "NoSleep" Writing Prompt Generator
Finetuned version of [GPT2](https://huggingface.co/gpt2) to facilitate generation of Writing Prompts for the [GPT-NoSleep-355m model](https://huggingface.co/DarwinAnim8or/GPT-NoSleep-355m)
You can use the space linked on the right to use this model, then use the NoSleep model in tandem to generate stories!
# Training Procedure
This was trained on the 'reddt-nosleep-posts' dataset, using the "HappyTransformers" library on Google Colab.
This model was trained for X epochs with learning rate 1e-2.
# Biases & Limitations
This likely contains the same biases and limitations as the original GPT2 that it is based on, and additionally heavy biases from the dataset.
It likely will generate offensive output.
# Intended Use
This model is meant for fun, nothing else.
# Sample Use
```python
from happytransformer import GENSettings
args_top_k = GENSettings(no_repeat_ngram_size=1, do_sample=True, top_k=80, temperature=0.4, max_length=25, early_stopping=True)
result = happy_gen.generate_text("[WP] \"", args=args_top_k)
print(result.text)
``` |