refact ouptut formats

auto_round/export/export_to_autoround/__init__.py

@@ -12,4 +12,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from .export import save_quantized_as_autoround, AutoRoundFormat
+from .export import save_quantized_as_autoround, AutoRoundExportFormat

auto_round/export/export_to_autoround/export.py

@@ -45,7 +45,7 @@
 )
 
 
-class AutoRoundFormat(str, Enum):
+class AutoRoundExportFormat(str, Enum):
     # Weight: FP8, per-channel, may be extended to per-tensor in future
     # Activation: FP8, per-tensor
     FP8_STATIC = "fp8_static"
@@ -165,8 +165,8 @@ def pack_layer(layer_name, model, backend, device=None):
         return pack_layer(layer_name, model, backend, device)
 
     if (
-        backend == f"auto_round:{AutoRoundFormat.FP8.value}"
-        or backend == f"auto_round:{AutoRoundFormat.FP8_STATIC.value}"
+        backend == f"auto_round:{AutoRoundExportFormat.FP8.value}"
+        or backend == f"auto_round:{AutoRoundExportFormat.FP8_STATIC.value}"
     ):
         from auto_round.export.export_to_autoround.export_to_fp8 import pack_layer
 
@@ -298,7 +298,7 @@ def save_quantized_as_autoround(output_dir, inplace=True, backend="auto_round:ex
     if (
         (kwargs.get("sym") is None or kwargs.get("sym"))
         and ("gptq" not in backend and "awq" not in backend)
-        and (AutoRoundFormat.FP8_STATIC.value not in backend)
+        and (AutoRoundExportFormat.FP8_STATIC.value not in backend)
     ):
         backend = backend.replace("auto_round", "auto_round:auto_gptq")