d-rau commited on
Commit
8ec008b
·
verified ·
1 Parent(s): c1d110c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -124,7 +124,7 @@ inputs = processor.apply_chat_template(
124
 
125
  outputs = model.generate(
126
  **inputs,
127
- max_new_tokens=128,
128
  )
129
 
130
  input_length = inputs["input_ids"].shape[-1]
 
124
 
125
  outputs = model.generate(
126
  **inputs,
127
+ max_new_tokens=32,
128
  )
129
 
130
  input_length = inputs["input_ids"].shape[-1]