Skip to content

Commit 3c12b86

Browse files
committed
format
1 parent 31b0460 commit 3c12b86

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

tests/openvino/test_quantization.py

+1-4
Original file line numberDiff line numberDiff line change
@@ -328,10 +328,7 @@ def test_ovmodel_4bit_weight_compression(self, model_cls, model_name, expected_i
328328

329329
quantizer = OVQuantizer.from_pretrained(transformers_model, task=task)
330330
ov_config = OVConfig(quantization_config=OVWeightQuantizationConfig(bits=4, sym=True, ratio=0.8))
331-
quantizer.quantize(
332-
save_directory=tmp_dir,
333-
ov_config=ov_config,
334-
)
331+
quantizer.quantize(save_directory=tmp_dir, ov_config=ov_config)
335332
model = model_cls.from_pretrained(tmp_dir)
336333

337334
_, num_int8, num_int4 = get_num_quantized_nodes(model)

0 commit comments

Comments
 (0)