Update README.md
#2
by
leons-esposa02 - opened
README.md
CHANGED
|
@@ -1,43 +1,53 @@
|
|
| 1 |
-
---
|
| 2 |
-
language: en
|
| 3 |
-
license: mit
|
| 4 |
-
pipeline_tag: text-generation
|
| 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 |
-
|
| 41 |
-
|
| 42 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 43 |
```
|
|
|
|
| 1 |
+
---
|
| 2 |
+
language: en
|
| 3 |
+
license: mit
|
| 4 |
+
pipeline_tag: text-generation
|
| 5 |
+
datasets:
|
| 6 |
+
- fka/awesome-chatgpt-prompts
|
| 7 |
+
metrics:
|
| 8 |
+
- accuracy
|
| 9 |
+
library_name: adapter-transformers
|
| 10 |
+
tags:
|
| 11 |
+
- art
|
| 12 |
+
- legal
|
| 13 |
+
- code
|
| 14 |
+
- text-generation-inference
|
| 15 |
+
---
|
| 16 |
+
# GPT-Neo 1.3B - Adventure
|
| 17 |
+
## Model Description
|
| 18 |
+
GPT-Neo 1.3B-Adventure is a finetune created using EleutherAI's GPT-Neo 1.3B model.
|
| 19 |
+
## Training data
|
| 20 |
+
The training data is a direct copy of the "cys" dataset by VE, a CYOA-based dataset.
|
| 21 |
+
### How to use
|
| 22 |
+
You can use this model directly with a pipeline for text generation. This example generates a different sequence each time it's run:
|
| 23 |
+
```py
|
| 24 |
+
>>> from transformers import pipeline
|
| 25 |
+
>>> generator = pipeline('text-generation', model='KoboldAI/GPT-Neo-1.3B-Adventure')
|
| 26 |
+
>>> generator("> You wake up.", do_sample=True, min_length=50)
|
| 27 |
+
[{'generated_text': '> You wake up"\nYou get out of bed, don your armor and get out of the door in search for new adventures.'}]
|
| 28 |
+
```
|
| 29 |
+
### Limitations and Biases
|
| 30 |
+
GPT-Neo was trained as an autoregressive language model. This means that its core functionality is taking a string of text and predicting the next token. While language models are widely used for tasks other than this, there are a lot of unknowns with this work.
|
| 31 |
+
GPT-Neo was trained on the Pile, a dataset known to contain profanity, lewd, and otherwise abrasive language. Depending on your usecase GPT-Neo may produce socially unacceptable text. See Sections 5 and 6 of the Pile paper for a more detailed analysis of the biases in the Pile.
|
| 32 |
+
As with all language models, it is hard to predict in advance how GPT-Neo will respond to particular prompts and offensive content may occur without warning. We recommend having a human curate or filter the outputs before releasing them, both to censor undesirable content and to improve the quality of the results.
|
| 33 |
+
### BibTeX entry and citation info
|
| 34 |
+
The model is made using the following software:
|
| 35 |
+
```bibtex
|
| 36 |
+
@software{gpt-neo,
|
| 37 |
+
author = {Black, Sid and
|
| 38 |
+
Leo, Gao and
|
| 39 |
+
Wang, Phil and
|
| 40 |
+
Leahy, Connor and
|
| 41 |
+
Biderman, Stella},
|
| 42 |
+
title = {{GPT-Neo: Large Scale Autoregressive Language
|
| 43 |
+
Modeling with Mesh-Tensorflow}},
|
| 44 |
+
month = mar,
|
| 45 |
+
year = 2021,
|
| 46 |
+
note = {{If you use this software, please cite it using
|
| 47 |
+
these metadata.}},
|
| 48 |
+
publisher = {Zenodo},
|
| 49 |
+
version = {1.0},
|
| 50 |
+
doi = {10.5281/zenodo.5297715},
|
| 51 |
+
url = {https://doi.org/10.5281/zenodo.5297715}
|
| 52 |
+
}
|
| 53 |
```
|