Skip to content

Commit d88c037

Browse files
committed
fix pattern
1 parent 3c77176 commit d88c037

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

tests/openvino/test_quantization.py

+7-7
Original file line numberDiff line numberDiff line change
@@ -91,7 +91,7 @@
9191
"text-classification": ("glue", "sst2", "sentence"),
9292
}
9393

94-
94+
pattern_prefix = "^__module.model.model" if is_transformers_version(">=", "4.49") else "^__module.model"
9595
class OVQuantizerTest(unittest.TestCase):
9696
SUPPORTED_ARCHITECTURES_TORCH_MODEL = (
9797
(OVModelForSequenceClassification, "bert", 32, 35),
@@ -158,12 +158,12 @@ class OVQuantizerTest(unittest.TestCase):
158158
dtype="nf4",
159159
group_size=16,
160160
ratio=0.5,
161-
ignored_scope={"patterns": ["^__module.model.model.layers.0.self_attn"]},
161+
ignored_scope={"patterns": [f"{pattern_prefix}.layers.0.self_attn"]},
162162
),
163163
full_quantization_config=OVQuantizationConfig(
164-
dtype="f8e4m3", ignored_scope={"patterns": ["^__module.model.model.layers.0.mlp"]}
164+
dtype="f8e4m3", ignored_scope={"patterns": [f"{pattern_prefix}.layers.0.mlp"]}
165165
),
166-
ignored_scope={"patterns": ["^__module.model.model.layers.1.self_attn"]},
166+
ignored_scope={"patterns": [f"{pattern_prefix}.layers.1.self_attn"]},
167167
dataset="wikitext2",
168168
num_samples=1,
169169
),
@@ -183,12 +183,12 @@ class OVQuantizerTest(unittest.TestCase):
183183
dtype="nf4",
184184
group_size=16,
185185
ratio=0.5,
186-
ignored_scope={"patterns": ["^__module.model.layers.0.self_attn"]},
186+
ignored_scope={"patterns": [f"{pattern_prefix}.layers.0.self_attn"]},
187187
),
188188
full_quantization_config=OVQuantizationConfig(
189-
dtype="f8e5m2", ignored_scope={"patterns": ["^__module.model.layers.0.mlp"]}
189+
dtype="f8e5m2", ignored_scope={"patterns": [f"{pattern_prefix}.layers.0.mlp"]}
190190
),
191-
ignored_scope={"patterns": ["^__module.model.layers.1.self_attn"]},
191+
ignored_scope={"patterns": [f"{pattern_prefix}.layers.1.self_attn"]},
192192
dataset="wikitext2",
193193
num_samples=1,
194194
),

0 commit comments

Comments
 (0)