Update README.md
Browse files
README.md
CHANGED
@@ -7,20 +7,20 @@ license: apache-2.0
|
|
7 |
tags:
|
8 |
- "text generation"
|
9 |
---
|
10 |
-
|
11 |
GPT-2 model from Lithuania using Wikipedia corpus dataset based on GPT-2 small model.
|
12 |
|
13 |
This is only the first version of the model, over time model will be improved using a bigger dataset and better data preparation.
|
14 |
|
15 |
-
|
16 |
This model was pre-trained with 180MB of Lithuanian Wikipedia. The texts are tokenized using a byte-level version of Byte Pair Encoding (BPE).
|
17 |
|
18 |
-
|
19 |
The model was trained on wiki-corpus for 40 hours using NVIDIA Tesla P100 GPU.
|
20 |
|
21 |
-
|
22 |
|
23 |
-
|
24 |
|
25 |
``` from transformers import AutoTokenizer, TFAutoModelWithLMHead
|
26 |
import tensorflow as tf
|
@@ -47,11 +47,11 @@ outputs = model.generate(inputs, eos_token_id=50256, pad_token_id=50256,
|
|
47 |
print(tokenizer.decode(outputs[0]))
|
48 |
|
49 |
```
|
50 |
-
|
51 |
The training data used for this model come from Lithuanian Wikipedia. We know it contains a lot of unfiltered content from the internet, which is far from neutral. As the openAI team themselves point out in their model card:
|
52 |
|
53 |
"Because large-scale language models like GPT-2 do not distinguish fact from fiction, we don’t support use-cases that require the generated text to be true. Additionally, language models like GPT-2 reflect the biases inherent to the systems they were trained on, so we do not recommend that they be deployed into systems that interact with humans > unless the deployers first carry out a study of biases relevant to the intended use-case. We found no statistically significant difference in gender, race, and religious bias probes between 774M and 1.5B, implying all versions of GPT-2 should be approached with similar levels of caution around use cases that are sensitive to biases around human attributes."
|
54 |
|
55 |
-
|
56 |
|
57 |
Lithuanian GPT-2 small was trained and evaluated by Deividas Mataciunas
|
|
|
7 |
tags:
|
8 |
- "text generation"
|
9 |
---
|
10 |
+
## Model description
|
11 |
GPT-2 model from Lithuania using Wikipedia corpus dataset based on GPT-2 small model.
|
12 |
|
13 |
This is only the first version of the model, over time model will be improved using a bigger dataset and better data preparation.
|
14 |
|
15 |
+
## Training data
|
16 |
This model was pre-trained with 180MB of Lithuanian Wikipedia. The texts are tokenized using a byte-level version of Byte Pair Encoding (BPE).
|
17 |
|
18 |
+
## Training
|
19 |
The model was trained on wiki-corpus for 40 hours using NVIDIA Tesla P100 GPU.
|
20 |
|
21 |
+
## How to use
|
22 |
|
23 |
+
### Load model
|
24 |
|
25 |
``` from transformers import AutoTokenizer, TFAutoModelWithLMHead
|
26 |
import tensorflow as tf
|
|
|
47 |
print(tokenizer.decode(outputs[0]))
|
48 |
|
49 |
```
|
50 |
+
## Limitations and bias
|
51 |
The training data used for this model come from Lithuanian Wikipedia. We know it contains a lot of unfiltered content from the internet, which is far from neutral. As the openAI team themselves point out in their model card:
|
52 |
|
53 |
"Because large-scale language models like GPT-2 do not distinguish fact from fiction, we don’t support use-cases that require the generated text to be true. Additionally, language models like GPT-2 reflect the biases inherent to the systems they were trained on, so we do not recommend that they be deployed into systems that interact with humans > unless the deployers first carry out a study of biases relevant to the intended use-case. We found no statistically significant difference in gender, race, and religious bias probes between 774M and 1.5B, implying all versions of GPT-2 should be approached with similar levels of caution around use cases that are sensitive to biases around human attributes."
|
54 |
|
55 |
+
## Author
|
56 |
|
57 |
Lithuanian GPT-2 small was trained and evaluated by Deividas Mataciunas
|