pop lm_head in recipe dict

xinhe3 · xinhe3 · commit 62f81e562666 · 2025-08-24T12:10:36.000+03:00
Signed-off-by: xinhe3 &lt;xinhe3@habana.ai&gt;
diff --git a/auto_round/autoround.py b/auto_round/autoround.py
@@ -3006,6 +3006,7 @@ def _generate_recipe(
         recipe_layer_config.update(self.recipe_results)
         self._dump_average_bits(layer_config=recipe_layer_config)
         self.recipe_mode = False
+        recipe_layer_config.pop("lm_head")  # lm_head is not included in the recipe
         return recipe_layer_config
 
     def _generate_block_recipe(self, block, input_ids, input_others):
