disable qbits lib import when using gpu version itrex

intel · Jun 14, 2024 · 317194c · 317194c
1 parent 127692d
commit 317194c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/intel_extension_for_transformers/qbits/__init__.py b/intel_extension_for_transformers/qbits/__init__.py
@@ -16,4 +16,5 @@
 # limitations under the License.
 
 import torch
-from intel_extension_for_transformers.qbits_py import * # pylint: disable=E0401, E0611
+if not torch.xpu._is_compiled():
+    from intel_extension_for_transformers.qbits_py import * # pylint: disable=E0401, E0611