From 68837f850d509dd932dc2c5eb91c76318681e7d2 Mon Sep 17 00:00:00 2001 From: ZX-ModelCloud Date: Mon, 16 Dec 2024 11:20:43 +0000 Subject: [PATCH] remove device check --- gptqmodel/nn_modules/qlinear/ipex.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/gptqmodel/nn_modules/qlinear/ipex.py b/gptqmodel/nn_modules/qlinear/ipex.py index e7802ed57..c950e5864 100644 --- a/gptqmodel/nn_modules/qlinear/ipex.py +++ b/gptqmodel/nn_modules/qlinear/ipex.py @@ -138,9 +138,6 @@ def validate(cls, bits: int, group_size: int, desc_act: bool, sym: bool, infeatu if sys.platform != "linux": return False, Exception("IPEX is only available on Linux platform.") - if device not in [DEVICE.CPU, DEVICE.XPU]: - return False, Exception(f"IPEX not supported on device: `{device}`.") - if not HAS_IPEX: return False, IPEX_ERROR_LOG return cls._validate(bits=bits, group_size=group_size, desc_act=desc_act, sym=sym, dynamic=dynamic, device=device, trainable=trainable)