fix bug for BaseDataset without init processor (#121)

ModelTC · Oct 11, 2024 · 6b04ac6 · 6b04ac6
1 parent 14e74a9
commit 6b04ac6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/llmc/data/dataset/base_dataset.py b/llmc/data/dataset/base_dataset.py
@@ -11,7 +11,7 @@
 
 
 class BaseDataset(metaclass=ABCMeta):
-    def __init__(self, tokenizer, calib_cfg, processor):
+    def __init__(self, tokenizer, calib_cfg, processor=None):
         # calib_cfg
         logger.info(f'calib_cfg : {calib_cfg}')
         self.tokenizer = tokenizer