Skip to content

Commit c2f373f

Browse files
committed
Fixed tests
1 parent 2ec2a54 commit c2f373f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/openvino/test_quantization.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -183,7 +183,7 @@ class OVWeightCompressionTest(unittest.TestCase):
183183
),
184184
(
185185
OVModelForCausalLM,
186-
"hf-internal-testing/tiny-random-gpt2",
186+
"hf-internal-testing/tiny-random-OPTForCausalLM",
187187
dict(
188188
mode=nncf.CompressWeightsMode.INT4_SYM,
189189
group_size=-1,
@@ -195,7 +195,7 @@ class OVWeightCompressionTest(unittest.TestCase):
195195
),
196196
(
197197
OVModelForCausalLM,
198-
"hf-internal-testing/tiny-random-gpt2",
198+
"hf-internal-testing/tiny-random-OPTForCausalLM",
199199
dict(
200200
mode=nncf.CompressWeightsMode.INT4_SYM,
201201
group_size=-1,

0 commit comments

Comments
 (0)