diff --git a/gptqmodel/looper/qqq_processor.py b/gptqmodel/looper/qqq_processor.py index b1f327f48..0b96be037 100644 --- a/gptqmodel/looper/qqq_processor.py +++ b/gptqmodel/looper/qqq_processor.py @@ -27,11 +27,11 @@ class QQQProcessor(LoopProcessor): def __init__(self, tokenizer, qcfg: QuantizeConfig, calibration, prepare_dataset_func, - calibration_concat_size: Optional[int], batch_size: int, + calibration_concat_size: Optional[int], calibration_sort: Optional[str], batch_size: int, logger_board: str = "", require_fwd: bool = True, calculate_w_wq_diff: bool = False): super().__init__(tokenizer=tokenizer, qcfg=qcfg, calibration=calibration, - calibration_concat_size=calibration_concat_size, + calibration_concat_size=calibration_concat_size, calibration_sort=calibration_sort, prepare_dataset_func=prepare_dataset_func, batch_size=batch_size, logger_board=logger_board, require_fwd=require_fwd)