skip gate

HDCharles · HDCharles · commit 80c344e1a9d9 · 2025-11-14T04:35:10.000Z
Summary

Signed-off-by: HDCharles &lt;charlesdavidhernandez@gmail.com&gt;
diff --git a/tests/e2e/e2e_utils.py b/tests/e2e/e2e_utils.py
@@ -84,11 +84,11 @@ def data_collator(batch):
                 targets="Linear",
                 scheme=scheme,
                 actorder=None,  # added for consistency with past testing configs
-                ignore=["lm_head"],
+                ignore=["lm_head", "re:.*mlp.gate.*"],
             )
         else:
             oneshot_kwargs["recipe"] = QuantizationModifier(
-                targets="Linear", scheme=scheme, ignore=["lm_head"]
+                targets="Linear", scheme=scheme, ignore=["lm_head", "re:.*mlp.gate.*"]
             )
 
     # Apply quantization.

Original file line number	Diff line number	Diff line change
`@@ -84,11 +84,11 @@ def data_collator(batch):`
`84`	`84`	`targets="Linear",`
`85`	`85`	`scheme=scheme,`
`86`	`86`	`actorder=None, # added for consistency with past testing configs`
`87`		`- ignore=["lm_head"],`
	`87`	`+ ignore=["lm_head", "re:.mlp.gate."],`
`88`	`88`	`)`
`89`	`89`	`else:`
`90`	`90`	`oneshot_kwargs["recipe"] = QuantizationModifier(`
`91`		`- targets="Linear", scheme=scheme, ignore=["lm_head"]`
	`91`	`+ targets="Linear", scheme=scheme, ignore=["lm_head", "re:.mlp.gate."]`
`92`	`92`	`)`
`93`	`93`
`94`	`94`	`# Apply quantization.`