Geralt-Targaryen commited on
Commit
518e5d5
1 Parent(s): 8cb0727

update readme

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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.95, temperature=1)
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
  ```