yu-val-weiss commited on
Commit
f053fe4
·
1 Parent(s): b8756a1

add model max length

Browse files
Files changed (1) hide show
  1. blimp.py +3 -1
blimp.py CHANGED
@@ -185,7 +185,9 @@ class Blimp(evaluate.Metric):
185
  model.eval()
186
 
187
  tokenizer = AutoTokenizer.from_pretrained(
188
- model_id, trust_remote_code=trust_remote_code
 
 
189
  )
190
 
191
  # if batch_size > 1 (which generally leads to padding being required), and
 
185
  model.eval()
186
 
187
  tokenizer = AutoTokenizer.from_pretrained(
188
+ model_id,
189
+ trust_remote_code=trust_remote_code,
190
+ max_model_length=150,
191
  )
192
 
193
  # if batch_size > 1 (which generally leads to padding being required), and