Geralt-Targaryen
/

FantasyGPT-tiny

Text Generation

text-generation-inference

Model card Files Files and versions

Geralt-Targaryen commited on Apr 19, 2023

Commit

c302294

·

1 Parent(s): f277591

update readme

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -29,8 +29,8 @@ Example usage:
 ```python
 from transformers import AutoTokenizer, GPT2LMHeadModel
-tokenizer = AutoTokenizer.from_pretrained('Geralt-Targaryen/FantasyGPT')
-model = GPT2LMHeadModel.from_pretrained('Geralt-Targaryen/FantasyGPT')
 input_text = ["Daenerys kissed Gandalf, as the witcher hacked off Lord Voldemort's head with a brutal swing of Longclaw."]
 input_tokenized = tokenizer(input_text, return_tensors='pt')

 ```python
 from transformers import AutoTokenizer, GPT2LMHeadModel
+tokenizer = AutoTokenizer.from_pretrained('Geralt-Targaryen/FantasyGPT-tiny')
+model = GPT2LMHeadModel.from_pretrained('Geralt-Targaryen/FantasyGPT-tiny')
 input_text = ["Daenerys kissed Gandalf, as the witcher hacked off Lord Voldemort's head with a brutal swing of Longclaw."]
 input_tokenized = tokenizer(input_text, return_tensors='pt')