Geralt-Targaryen
commited on
Commit
•
518e5d5
1
Parent(s):
8cb0727
update readme
Browse files
README.md
CHANGED
@@ -35,7 +35,7 @@ model = GPT2LMHeadModel.from_pretrained('Geralt-Targaryen/FantasyGPT')
|
|
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')
|
37 |
|
38 |
-
output = model.generate(inputs=input_tokenized.input_ids, max_new_tokens=256, do_sample=True, top_p=0.
|
39 |
|
40 |
print(tokenizer.decode(output[0]))
|
41 |
```
|
|
|
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')
|
37 |
|
38 |
+
output = model.generate(inputs=input_tokenized.input_ids, max_new_tokens=256, do_sample=True, top_p=0.9, temperature=1)
|
39 |
|
40 |
print(tokenizer.decode(output[0]))
|
41 |
```
|