From d40fa3394461d0c855f3bab859a218404f3e544f Mon Sep 17 00:00:00 2001 From: Wenhua Cheng Date: Tue, 23 Dec 2025 14:58:17 +0800 Subject: [PATCH 1/2] update --- auto_round/schemes.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/auto_round/schemes.py b/auto_round/schemes.py index 56cf72b2e..8754c8dea 100644 --- a/auto_round/schemes.py +++ b/auto_round/schemes.py @@ -178,6 +178,18 @@ def is_preset_scheme(name: str) -> bool: ) MXFP8 = QuantizationScheme.from_dict( + { + "bits": 8, + "group_size": 32, + "data_type": "mx_fp", + "act_bits": 8, + "act_data_type": "mx_fp", + "act_group_size": 32, + "act_sym": True, + } +) + +MXFP8_RCEIL = QuantizationScheme.from_dict( { "bits": 8, "group_size": 32, @@ -189,6 +201,7 @@ def is_preset_scheme(name: str) -> bool: } ) + NVFP4 = QuantizationScheme.from_dict( { "bits": 4, @@ -251,6 +264,7 @@ def is_preset_scheme(name: str) -> bool: "W8A16": W8A16, "MXFP4": MXFP4, "MXFP8": MXFP8, + "MXFP8_RCEIL":MXFP8_RCEIL, "NVFP4": NVFP4, "FPW8A16": FPW8A16, "W2A16G64": W2A16G64, From c97f7caafc790400e70e29535fb3854d80818be6 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Tue, 23 Dec 2025 06:59:31 +0000 Subject: [PATCH 2/2] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- auto_round/schemes.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/auto_round/schemes.py b/auto_round/schemes.py index f76094eb3..0ca7710e0 100644 --- a/auto_round/schemes.py +++ b/auto_round/schemes.py @@ -278,7 +278,7 @@ def is_preset_scheme(name: str) -> bool: "MXFP4": MXFP4, "MXFP4_RCEIL": MXFP4_RCEIL, "MXFP8": MXFP8, - "MXFP8_RCEIL":MXFP8_RCEIL, + "MXFP8_RCEIL": MXFP8_RCEIL, "NVFP4": NVFP4, "FPW8A16": FPW8A16, "W2A16G64": W2A16G64,