Geralt-Targaryen commited on
Commit
c302294
·
1 Parent(s): f277591

update readme

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