ClementRomac HF Staff commited on
Commit
717f69a
·
1 Parent(s): d4c66eb

Upload processor

Browse files
Files changed (1) hide show
  1. processor.py +1 -1
processor.py CHANGED
@@ -13,7 +13,7 @@ class GIAProcessor(GitProcessor):
13
  }
14
  for i in range(len(examples["input_ids"])):
15
  _input_size = len(examples["input_ids"][i])
16
- for j in range(_input_size // self._block_size):
17
  results["input_ids"].append(examples["input_ids"][i][j*self._block_size:(j+1)*self._block_size])
18
  results["attention_mask"].append(examples["attention_mask"][i][j * self._block_size:(j + 1) * self._block_size])
19
 
 
13
  }
14
  for i in range(len(examples["input_ids"])):
15
  _input_size = len(examples["input_ids"][i])
16
+ for j in range(max(1, _input_size // self._block_size)):
17
  results["input_ids"].append(examples["input_ids"][i][j*self._block_size:(j+1)*self._block_size])
18
  results["attention_mask"].append(examples["attention_mask"][i][j * self._block_size:(j + 1) * self._block_size])
19