shunxing1234 commited on
Commit
8df699a
·
verified ·
1 Parent(s): 6657188

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -95,7 +95,7 @@ model_inputs = tokenizer([text], return_tensors="pt").to(model.device)
95
  # conduct text completion
96
  generated_ids = model.generate(
97
  **model_inputs,
98
- max_new_tokens=40960,
99
  temperature=0.6,
100
  top_p=0.95,
101
  )
 
95
  # conduct text completion
96
  generated_ids = model.generate(
97
  **model_inputs,
98
+ max_new_tokens=32768,
99
  temperature=0.6,
100
  top_p=0.95,
101
  )