XXXiong commited on
Commit
72af62e
·
verified ·
1 Parent(s): 82046e5

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -139,7 +139,7 @@ model_inputs = tokenizer([text], return_tensors="pt").to(model.device)
139
 
140
  generated_ids = model.generate(
141
  **model_inputs,
142
- max_new_tokens=1024,
143
  temperature=0.7
144
  )
145
  generated_ids = [
 
139
 
140
  generated_ids = model.generate(
141
  **model_inputs,
142
+ max_new_tokens=2048,
143
  temperature=0.7
144
  )
145
  generated_ids = [