fix

huggingface · Oct 16, 2023 · 7a08831 · 7a08831
1 parent c65b36a
commit 7a08831
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/optimum/intel/utils/import_utils.py b/optimum/intel/utils/import_utils.py
@@ -17,7 +17,6 @@
 import sys
 from collections import OrderedDict
 from typing import Union
-
 from packaging.version import Version, parse
 
 
@@ -205,7 +204,10 @@ def is_torch_version(operation: str, version: str):
     """
     if not _torch_available:
         return False
-    return compare_versions(parse(_torch_version), operation, version)
+
+    import torch 
+
+    return compare_versions(parse(parse(torch.__version__).base_version), operation, version)
 
 
 def is_ipex_version(operation: str, version: str):