Skip to content

Commit 6b04ac6

Browse files
fix bug for BaseDataset without init processor (#121)
1 parent 14e74a9 commit 6b04ac6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

llmc/data/dataset/base_dataset.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111

1212

1313
class BaseDataset(metaclass=ABCMeta):
14-
def __init__(self, tokenizer, calib_cfg, processor):
14+
def __init__(self, tokenizer, calib_cfg, processor=None):
1515
# calib_cfg
1616
logger.info(f'calib_cfg : {calib_cfg}')
1717
self.tokenizer = tokenizer

0 commit comments

Comments
 (0)