andrewkroening commited on
Commit
05e1e8a
1 Parent(s): ecb2778

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +60 -0
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)