hikahika commited on
Commit
c54c1f4
·
verified ·
1 Parent(s): 6fb4678

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -79,7 +79,7 @@ for data in tqdm(datasets):
79
  with torch.no_grad():
80
  outputs = model.generate(
81
  tokenized_input,
82
- max_new_tokens=100,
83
  do_sample=False,
84
  repetition_penalty=1.2
85
  )[0]
 
79
  with torch.no_grad():
80
  outputs = model.generate(
81
  tokenized_input,
82
+ max_new_tokens=512,
83
  do_sample=False,
84
  repetition_penalty=1.2
85
  )[0]