ekrombouts commited on
Commit
1b378d5
·
verified ·
1 Parent(s): b88552b

Update README.md

Browse files

increased max_new_tokens in example code

Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -61,7 +61,7 @@ Antwoord:
61
  """
62
 
63
  input_ids = tokenizer(prompt, return_tensors="pt").input_ids
64
- output = model.generate(input_ids)
65
  print(tokenizer.decode(output[0], skip_special_tokens=True))
66
  ```
67
  ## Training Details
 
61
  """
62
 
63
  input_ids = tokenizer(prompt, return_tensors="pt").input_ids
64
+ output = model.generate(input_ids, max_new_tokens=1024))
65
  print(tokenizer.decode(output[0], skip_special_tokens=True))
66
  ```
67
  ## Training Details