Skip to content

Commit d41eac3

Browse files
authored
revert typo
1 parent cc13a65 commit d41eac3

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

auto_round_extension/qbits/qlinear_qbits_gptq.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -92,6 +92,14 @@ def req_check(self):
9292
torch_version = str(torch.__version__)
9393
if QBITS_AVAILABLE:
9494
pass
95+
# import intel_extension_for_transformers
96+
# itrex_version = str(intel_extension_for_transformers.__version__)
97+
# version_match_map = {"1.4": "2.2.0+cpu",
98+
# "1.4.1": "2.2.0+cpu", "1.4.2": "2.3.0+cpu"}
99+
# if itrex_version in version_match_map:
100+
# if torch_version != version_match_map[itrex_version]:
101+
# logger.warning(
102+
# f"Please install torch {version_match_map[itrex_version]} by command 'pip install torch=={version_match_map[itrex_version]} --extra-index-url https://download.pytorch.org/whl/cpu' as Intel Extension for Transformers {itrex_version} is not compatible with current torch.")
95103
else:
96104
logger.error(
97105
"Please install Intel Extension for Transformers by running 'pip install intel-extension-for-transformers' as qbits linear requirements checking fail. "

0 commit comments

Comments
 (0)