XXXiong commited on
Commit
427dd4a
·
verified ·
1 Parent(s): 7db33bb

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -155,7 +155,7 @@ model_inputs = tokenizer([text], return_tensors="pt").to(model.device)
155
 
156
  generated_ids = model.generate(
157
  **model_inputs,
158
- max_new_tokens=1024,
159
  temperature=0.7
160
  )
161
  generated_ids = [
 
155
 
156
  generated_ids = model.generate(
157
  **model_inputs,
158
+ max_new_tokens=2048,
159
  temperature=0.7
160
  )
161
  generated_ids = [