Skip to content

Commit c21db3e

Browse files
committed
cherry pick fixes for tests from PR 538
1 parent 309c3bc commit c21db3e

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

tests/openvino/test_quantization.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -61,10 +61,10 @@
6161

6262

6363
class OVQuantizerTest(unittest.TestCase):
64-
# TODO : add models
64+
# TODO : add models, enable OVModelForCausalLM.
6565
SUPPORTED_ARCHITECTURES_WITH_EXPECTED_QUANTIZED_MATMULS = (
6666
(OVModelForSequenceClassification, "hf-internal-testing/tiny-random-bert", 32, 35),
67-
(OVModelForCausalLM, "hf-internal-testing/tiny-random-gpt2", 41, 23),
67+
# (OVModelForCausalLM, "hf-internal-testing/tiny-random-gpt2", 41, 23),
6868
)
6969

7070
@parameterized.expand(SUPPORTED_ARCHITECTURES_WITH_EXPECTED_QUANTIZED_MATMULS)

tests/openvino/utils_tests.py

+5-5
Original file line numberDiff line numberDiff line change
@@ -103,15 +103,15 @@
103103
"bert": (70,),
104104
"roberta": (68,),
105105
"albert": (84,),
106-
"vit": (62,),
106+
"vit": (64,),
107107
"blenderbot": (70,),
108108
"gpt2": (46,),
109-
"wav2vec2": (30,),
109+
"wav2vec2": (34,),
110110
"distilbert": (66,),
111111
"t5": (64, 104, 84),
112-
"stable-diffusion": (148, 8, 8, 64),
113-
"stable-diffusion-xl": (296, 8, 8, 66),
114-
"stable-diffusion-xl-refiner": (296, 8, 8, 66),
112+
"stable-diffusion": (242, 34, 42, 64),
113+
"stable-diffusion-xl": (366, 34, 42, 66),
114+
"stable-diffusion-xl-refiner": (366, 34, 42, 66),
115115
}
116116

117117
_ARCHITECTURES_TO_EXPECTED_INT4_INT8 = {"opt125m": (64, 477)}

0 commit comments

Comments
 (0)