fix any_model for gpt-oss

danielkorzekwa · danielkorzekwa · commit 153847ccbe34 · 2026-01-29T09:58:05.000-08:00
Signed-off-by: Daniel Korzekwa &lt;dkorzekwa@nvidia.com&gt;
diff --git a/tests/gpu/torch/puzzletron/resources/configs/gpt-oss-20b/pruning/expert_removal.yaml b/tests/gpu/torch/puzzletron/resources/configs/gpt-oss-20b/pruning/expert_removal.yaml
@@ -17,3 +17,4 @@ num_experts_to_keep_list: [24, 16, 8]
 mlp_init_mode: "ExpertRemoval"
 mlp_init_config_yaml:
   expert_scores_key: "expert_ranks_mse"
+  layer_prefix_template: "model.layers.{layer_idx}."
