We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 4a1cf6c commit 1e17e05Copy full SHA for 1e17e05
optimum/intel/ipex/modeling_base.py
@@ -54,7 +54,7 @@
54
_patch_model,
55
)
56
from ..utils.constant import _TASK_ALIASES
57
-from ..utils.import_utils import is_ipex_version, is_transformers_version
+from ..utils.import_utils import is_ipex_version, is_torch_version, is_transformers_version
58
from ..utils.modeling_utils import recursive_to_device
59
60
0 commit comments