ronald commited on
Commit
4ebdc9c
·
1 Parent(s): 17201a4
Files changed (1) hide show
  1. local_coh_ppl.py +1 -0
local_coh_ppl.py CHANGED
@@ -211,6 +211,7 @@ class LocalCohPPL(evaluate.Measurement):
211
  )
212
 
213
  # tokenize by sentence
 
214
  for pred in batch_sents:
215
  ss = pred.split("\n")
216
  sslens = [len(tokenizer(y,add_special_tokens=False,padding=False).input_ids) for y in ss]
 
211
  )
212
 
213
  # tokenize by sentence
214
+ sent_tok_lens = []
215
  for pred in batch_sents:
216
  ss = pred.split("\n")
217
  sslens = [len(tokenizer(y,add_special_tokens=False,padding=False).input_ids) for y in ss]