fix a bug

intel · May 31, 2024 · debc852 · debc852
1 parent 5025bd4
commit debc852
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/auto_round/export/export_to_autoround/export.py b/auto_round/export/export_to_autoround/export.py
@@ -96,7 +96,7 @@ def dynamic_QuantLienar_for_packing(backend, bits, group_size):
         assert False, f"only support gptq and autoround backend"
 
 
-@register_format("autoround")
+@register_format("auto_round")
 def save_quantized_as_autoround(output_dir, inplace=True, backend="autoround:exllamav2", **kwargs):
     model = kwargs["model"]
     if not inplace: