redscroll commited on
Commit
3ed7103
1 Parent(s): 14451fd

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -16,7 +16,7 @@ input_text = "In my younger and more vulnerable years"
16
 
17
  input_ids = tokenizer(input_text, return_tensors = "pt").to("cuda")
18
 
19
- outputs = model.generate(**input_ids, max_new_tokens = 500, pad_token_id=tokenizer.eos_token_id, eos_token_id=tokenizer.eos_token_id, do_sample=True)
20
 
21
  print(tokenizer.decode(outputs[0]))
22
  ```
 
16
 
17
  input_ids = tokenizer(input_text, return_tensors = "pt").to("cuda")
18
 
19
+ outputs = model.generate(**input_ids, max_new_tokens = 500, pad_token_id=tokenizer.eos_token_id, eos_token_id=tokenizer.eos_token_id)
20
 
21
  print(tokenizer.decode(outputs[0]))
22
  ```