Skip to content
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.

Commit a297f09

Browse files
committedFeb 29, 2024·
fix
1 parent aa667ef commit a297f09

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed
 

‎tests/openvino/test_quantization.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -332,7 +332,7 @@ def test_ovmodel_8bit_weight_compression_stateful(self, model_cls, model_id, exp
332332
@parameterized.expand(SUPPORTED_ARCHITECTURES_WITH_AUTO_COMPRESSION)
333333
def test_ovmodel_load_with_compressed_weights(self, model_cls, model_type):
334334
model = model_cls.from_pretrained(MODEL_NAMES[model_type], export=True, load_in_8bit=True, stateful=False)
335-
self.assertEqual(model._openvino_config.quantization_config["bits"], 8)
335+
self.assertEqual(model._openvino_config.quantization_config.bits, 8)
336336
self.assertEqual(model._openvino_config.dtype, "int8")
337337

338338
if model.export_feature.startswith("text2text-generation"):

0 commit comments

Comments
 (0)
Please sign in to comment.