62
62
OVTrainer ,
63
63
OVQuantizationConfig ,
64
64
OVWeightQuantizationConfig ,
65
+ OVDynamicQuantizationConfig ,
65
66
)
66
67
from optimum .intel .openvino .configuration import OVQuantizationMethod , OVQuantizationConfigBase
67
68
@@ -684,12 +685,7 @@ class OVQuantizationConfigTest(unittest.TestCase):
684
685
QUANTIZATION_CONFIGS = (
685
686
(None ,),
686
687
(OVWeightQuantizationConfig (),),
687
- (
688
- OVWeightQuantizationConfig (
689
- bits = 8 ,
690
- sym = True ,
691
- ),
692
- ),
688
+ (OVWeightQuantizationConfig (bits = 8 ,sym = True ),),
693
689
(
694
690
OVWeightQuantizationConfig (
695
691
dataset = "wikitext" ,
@@ -717,6 +713,7 @@ class OVQuantizationConfigTest(unittest.TestCase):
717
713
),
718
714
),
719
715
(OVQuantizationConfig (ignored_scope = nncf .IgnoredScope (names = ["op_name" ])),),
716
+ (OVDynamicQuantizationConfig (bits = 8 ,sym = True ),),
720
717
)
721
718
722
719
QUANTIZATION_CONFIG_DICTS = (
@@ -746,11 +743,7 @@ class OVQuantizationConfigTest(unittest.TestCase):
746
743
),
747
744
(dict (num_samples = 100 ), OVWeightQuantizationConfig , "Can't determine type of OV quantization config" ),
748
745
(dict (abc = "def" ), OVWeightQuantizationConfig , "Can't determine type of OV quantization config" ),
749
- (
750
- dict (bits = 8 , fast_bias_correction = True ),
751
- OVWeightQuantizationConfig ,
752
- "Can't determine type of OV quantization config" ,
753
- ),
746
+ (dict (bits = 8 , fast_bias_correction = True , dataset = "wikitext" ), OVWeightQuantizationConfig , "Can't determine type of OV quantization config" ),
754
747
(dict (model_type = "transformer" ), OVQuantizationConfig , None ),
755
748
(
756
749
dict (
@@ -768,7 +761,7 @@ class OVQuantizationConfigTest(unittest.TestCase):
768
761
(dict (weight_only = False ), OVQuantizationConfig , None ),
769
762
(dict (abc = "def" , weight_only = False ), OVQuantizationConfig , None ),
770
763
(dict (abc = "def" , weight_only = True ), OVWeightQuantizationConfig , None ),
771
- (dict (bits = 8 , fast_bias_correction = True , weight_only = True ), OVWeightQuantizationConfig , None ),
764
+ (dict (bits = 8 , fast_bias_correction = True , dataset = "wikitext" , weight_only = True ), OVWeightQuantizationConfig , None ),
772
765
(dict (bits = 8 , fast_bias_correction = True , weight_only = False ), OVQuantizationConfig , None ),
773
766
)
774
767
0 commit comments