Skip to content

Commit ac9aee8

Browse files
committed
Fixed UT error
Signed-off-by: Cheng, Penghui <penghui.cheng@intel.com>
1 parent 721dd3b commit ac9aee8

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/neural_compressor/test_optimization.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -240,8 +240,9 @@ def test_weight_only_quantization(self, no_config, algo, weight_dtype):
240240
)
241241
else:
242242
q_model = quantizer.quantize(
243-
weight_only=True, # use RTN quantization method and NF4 weight data type is default.
243+
quantization_config=None,
244244
save_directory=tmp_dir,
245+
weight_only=True, # use RTN quantization method and NF4 weight data type is default.
245246
)
246247
q_model = INCModelForCausalLM.from_pretrained(tmp_dir)
247248
inp = torch.tensor([calibration_dataset[0]["input_ids"]])

0 commit comments

Comments
 (0)