PartiallyTyped
commited on
Commit
·
20e1b27
1
Parent(s):
878b02f
Update README.md
Browse files
README.md
CHANGED
@@ -2,4 +2,12 @@
|
|
2 |
|
3 |
This is a pretrained model based on [gpt2](https://huggingface.co/gpt2) that has been trained on [copenlu/answerable_tydiqa](https://huggingface.co/datasets/copenlu/answerable_tydiqa), specifically the text field of the English samples for 2 epochs.
|
4 |
|
5 |
-
To use the pretrained head, use: `AutoModelForCausalLM.from_pretrained`.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
2 |
|
3 |
This is a pretrained model based on [gpt2](https://huggingface.co/gpt2) that has been trained on [copenlu/answerable_tydiqa](https://huggingface.co/datasets/copenlu/answerable_tydiqa), specifically the text field of the English samples for 2 epochs.
|
4 |
|
5 |
+
To use the pretrained head, use: `AutoModelForCausalLM.from_pretrained`.
|
6 |
+
|
7 |
+
```python
|
8 |
+
from transformers import AutoModelForCausalLM, AutoTokenizer
|
9 |
+
|
10 |
+
model = AutoModelForCausalLM.from_pretrained("PartiallyTyped/answerable_tydiqa_lm_pretrained_japenese")
|
11 |
+
tokenizer = AutoTokenizer.from_pretrained("rinna/japanese-gpt2-small")
|
12 |
+
|
13 |
+
```
|