remove device check

ZX-ModelCloud · Dec 16, 2024 · 68837f8 · 68837f8
1 parent f24603c
commit 68837f8
Showing 1 changed file with 0 additions and 3 deletions.
diff --git 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)