Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -44,7 +44,7 @@ model.eval()
44
 
45
  prompt = 'Q: What is the largest bird?\nA:'
46
  input_ids = tokenizer(prompt, return_tensors="pt").input_ids
47
- tokens = model.generate(input_ids, max_length=20)
48
  print( tokenizer.decode(tokens[0].tolist(), skip_special_tokens=True) )
49
  # Q: What is the largest bird?\nA: The largest bird is a black-headed gull.
50
  ```
 
44
 
45
  prompt = 'Q: What is the largest bird?\nA:'
46
  input_ids = tokenizer(prompt, return_tensors="pt").input_ids
47
+ tokens = model.generate(input_ids, max_length=22)
48
  print( tokenizer.decode(tokens[0].tolist(), skip_special_tokens=True) )
49
  # Q: What is the largest bird?\nA: The largest bird is a black-headed gull.
50
  ```