Commit
·
717f69a
1
Parent(s):
d4c66eb
Upload processor
Browse files- 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 |
|