Commit
·
c302294
1
Parent(s):
f277591
update readme
Browse files
README.md
CHANGED
@@ -29,8 +29,8 @@ Example usage:
|
|
29 |
```python
|
30 |
from transformers import AutoTokenizer, GPT2LMHeadModel
|
31 |
|
32 |
-
tokenizer = AutoTokenizer.from_pretrained('Geralt-Targaryen/FantasyGPT')
|
33 |
-
model = GPT2LMHeadModel.from_pretrained('Geralt-Targaryen/FantasyGPT')
|
34 |
|
35 |
input_text = ["Daenerys kissed Gandalf, as the witcher hacked off Lord Voldemort's head with a brutal swing of Longclaw."]
|
36 |
input_tokenized = tokenizer(input_text, return_tensors='pt')
|
|
|
29 |
```python
|
30 |
from transformers import AutoTokenizer, GPT2LMHeadModel
|
31 |
|
32 |
+
tokenizer = AutoTokenizer.from_pretrained('Geralt-Targaryen/FantasyGPT-tiny')
|
33 |
+
model = GPT2LMHeadModel.from_pretrained('Geralt-Targaryen/FantasyGPT-tiny')
|
34 |
|
35 |
input_text = ["Daenerys kissed Gandalf, as the witcher hacked off Lord Voldemort's head with a brutal swing of Longclaw."]
|
36 |
input_tokenized = tokenizer(input_text, return_tensors='pt')
|