Skip to content

Commit 4842055

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 212b855 commit 4842055

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

auto_round/inference/backend.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -629,15 +629,13 @@ def dynamic_import_inference_linear(backend, config):
629629
if "torch_nvfp4" in backend:
630630
return ar_qmodules.NVFP4QuantLinear
631631

632-
if "auto_round_kernel" in backend or 'ark' in backend:
632+
if "auto_round_kernel" in backend or "ark" in backend:
633633
try:
634634
import auto_round_kernel as ark # pylint: disable=E0401
635635
except Exception as e:
636-
raise ImportError(
637-
"Please install auto_round_kernel version for CPU/XPU"
638-
)
636+
raise ImportError("Please install auto_round_kernel version for CPU/XPU")
639637
import auto_round_extension.kernel.qlinear as qlinear
640-
638+
641639
if "zp" in backend:
642640
return qlinear.QuantLinearGPTQ
643641
elif "awq" in backend:

0 commit comments

Comments
 (0)