andrewkroening
commited on
Commit
•
05e1e8a
1
Parent(s):
ecb2778
Create README.md
Browse files
README.md
ADDED
@@ -0,0 +1,60 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
language: en
|
3 |
+
tags:
|
4 |
+
- conversational
|
5 |
+
|
6 |
+
license: cc
|
7 |
+
---
|
8 |
+
|
9 |
+
|
10 |
+
# GPT-2
|
11 |
+
|
12 |
+
This model is based on a GPT-2 model which was fine-tuned on a Hugging Face dataset. It is intended largely as an illustrative example and is not intended to be used for any serious purpose. It's trained on a movie script for goodness' sake.
|
13 |
+
|
14 |
+
Disclaimer: The team releasing GPT-2 also wrote a
|
15 |
+
[model card](https://github.com/openai/gpt-2/blob/master/model_card.md) for their model. Content from this model card
|
16 |
+
has been written by the Hugging Face team to complete the information they provided and give specific examples of bias.
|
17 |
+
|
18 |
+
## Acknowledgements
|
19 |
+
|
20 |
+
There are several sources of inspiration and insight for the project that spawned this model. I'd like to recognize them up front:
|
21 |
+
|
22 |
+
* The [Microsoft DialoGPT-Medium](https://huggingface.co/microsoft/DialoGPT-medium?text=Hi.) model page was very insightful for getting stated.
|
23 |
+
|
24 |
+
* Lynn Zheng [r3dhummingbird](https://huggingface.co/r3dhummingbird/DialoGPT-medium-joshua?text=Hey+my+name+is+Thomas%21+How+are+you%3F) put together one heck of an awesome tutorial on how to fine-tune GPT-2 for conversational purposes. I used her tutorial as a starting point for this project. Check out the [Github repo here.](https://github.com/RuolinZheng08/twewy-discord-chatbot)
|
25 |
+
|
26 |
+
* [This article](https://towardsdatascience.com/make-your-own-rick-sanchez-bot-with-transformers-and-dialogpt-fine-tuning-f85e6d1f4e30) was also very insightful. Written by Rostyslav Neskorozhenyi.
|
27 |
+
|
28 |
+
* From a lineage standpoint, it looks like Nathan Cooper kicked this whole thing off with this [notebook.](https://github.com/ncoop57/i-am-a-nerd/blob/master/_notebooks/2020-05-12-chatbot-part-1.ipynb)
|
29 |
+
|
30 |
+
* Noah Gift figured out a few of the big pieces in [this repository.](https://github.com/nogibjj/hugging-face-tutorial-practice)
|
31 |
+
|
32 |
+
* I'd be remiss if I also didn't mention Hugging Face's own support [documentation](https://huggingface.co/transformers/v2.0.0/examples.html#gpt-2-gpt-and-causal-language-modeling) and team. All around great.
|
33 |
+
|
34 |
+
## Model description
|
35 |
+
|
36 |
+
This model uses GPT-2 Medium as a base model and was fine-tuned using scripts from the original (and best) Star Wars Trilogy. In this particular case, it was fine-tuned on Han Solo's 450-some lines. This is not a lot, and thus the model should not be assumed to have serious integrity. It's just a fun little project.
|
37 |
+
|
38 |
+
## Intended uses & limitations
|
39 |
+
|
40 |
+
This model is intended to be used for fun and entertainment. Don't take it too seriously.
|
41 |
+
|
42 |
+
### Ways to use
|
43 |
+
|
44 |
+
You can always chat with the model directly on the Hugging Face website. Just click the "Chat" button on the right side of the model page.
|
45 |
+
|
46 |
+
If you want to use the model in your own project, I recommend you train it better using much more data.
|
47 |
+
|
48 |
+
To access the GitHub repository I used to train this model, click [here](https://github.com/nogibjj/hugging-face-gpt-trainer/tree/gpt-fine-tune)
|
49 |
+
|
50 |
+
## Fine-tuning data
|
51 |
+
|
52 |
+
The script to generate this model takes a Hugging Face data set in this approximate format:
|
53 |
+
|
54 |
+
| Speaker | Text |
|
55 |
+
| --- | --- |
|
56 |
+
| Luke | Hello there. |
|
57 |
+
| Han | General Kenobi. |
|
58 |
+
| Luke | You are a bold one. |
|
59 |
+
|
60 |
+
The script then asks the user to define parameters for making the dataset and proceeding to fine-tuning. The actual dataset for this model can be found [here.](andrewkroening/Star-wars-scripts-dialogue-IV-VI)
|