kiranr commited on
Commit
9b9cebe
1 Parent(s): 7088235

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -46,7 +46,7 @@ model_inputs = tokenizer(input_text.format(prompt=prompt), return_tensors="pt").
46
  gen_conf = {
47
  "temperature": 0.7,
48
  "repetition_penalty": 1.0,
49
- "max_new_tokens": 100,
50
  "do_sample": True,
51
  }
52
 
 
46
  gen_conf = {
47
  "temperature": 0.7,
48
  "repetition_penalty": 1.0,
49
+ "max_new_tokens": 512,
50
  "do_sample": True,
51
  }
52