Skip to content

Commit ae1da0f

Browse files
committed
Fixed test
1 parent 4707914 commit ae1da0f

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

tests/openvino/test_quantization.py

+6-1
Original file line numberDiff line numberDiff line change
@@ -286,10 +286,15 @@ def test_ovmodel_4bit_weight_compression(self, model_cls, model_name, expected_i
286286
tokenizer.pad_token = tokenizer.eos_token
287287

288288
quantizer = OVQuantizer.from_pretrained(transformers_model, task=task)
289+
ov_config = OVConfig(
290+
weight_quantization_config=OVWeightQuantizationConfig(
291+
mode=nncf.CompressWeightsMode.INT4_SYM, ratio=0.8
292+
)
293+
)
289294
quantizer.quantize(
290295
save_directory=tmp_dir,
291296
weights_only=True,
292-
quantization_config=OVWeightQuantizationConfig(mode=nncf.CompressWeightsMode.INT4_SYM, ratio=0.8),
297+
ov_config=ov_config,
293298
)
294299
model = model_cls.from_pretrained(tmp_dir)
295300

0 commit comments

Comments
 (0)