fix

intel · Jun 27, 2024 · 2fa2c3f · 2fa2c3f
1 parent 655791e
commit 2fa2c3f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/intel_extension_for_transformers/qbits/__init__.py b/intel_extension_for_transformers/qbits/__init__.py
@@ -17,5 +17,5 @@
 
 import torch
 import intel_extension_for_transformers
-if "gpu" in intel_extension_for_transformers.__version__:
+if "gpu" not in intel_extension_for_transformers.__version__:
     from intel_extension_for_transformers.qbits_py import *  # pylint: disable=E0401, E0611