E6E831728 commited on
Commit
95bbbcc
·
verified ·
1 Parent(s): cbf1b13

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -79,7 +79,7 @@ prompt = "Question: What is the capital of UK?\nAnswer:"
79
  input_ids = torch.tensor([tokenizer.encode(prompt)], dtype=torch.long)
80
 
81
  with torch.no_grad():
82
- output_ids = model.generate(input_ids, max_new_tokens=16, do_sample=False)
83
 
84
  print(tokenizer.decode(output_ids[0].tolist()))
85
  ```
 
79
  input_ids = torch.tensor([tokenizer.encode(prompt)], dtype=torch.long)
80
 
81
  with torch.no_grad():
82
+ output_ids = model.generate(input_ids, max_new_tokens=3, do_sample=False)
83
 
84
  print(tokenizer.decode(output_ids[0].tolist()))
85
  ```