Skip to content

Commit 721dd3b

Browse files
committed
Changed parameters order in INCQuantizer.quantize
Signed-off-by: Cheng, Penghui <penghui.cheng@intel.com>
1 parent 5ddd360 commit 721dd3b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

optimum/intel/neural_compressor/quantization.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -143,8 +143,8 @@ def from_pretrained(cls, model: PreTrainedModel, **kwargs):
143143

144144
def quantize(
145145
self,
146+
quantization_config: Config,
146147
save_directory: Union[str, Path],
147-
quantization_config: Config = None,
148148
calibration_dataset: Dataset = None,
149149
batch_size: int = 8,
150150
data_collator: Optional[DataCollator] = None,

0 commit comments

Comments
 (0)