add missing import

echarlaix · echarlaix · commit 1e17e05ea6a2 · 2025-02-27T11:07:46.000+01:00
diff --git a/optimum/intel/ipex/modeling_base.py b/optimum/intel/ipex/modeling_base.py
@@ -54,7 +54,7 @@
     _patch_model,
 )
 from ..utils.constant import _TASK_ALIASES
-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
 from ..utils.modeling_utils import recursive_to_device
 
 

Original file line number	Diff line number	Diff line change
`@@ -54,7 +54,7 @@`
`54`	`54`	`_patch_model,`
`55`	`55`	`)`
`56`	`56`	`from ..utils.constant import _TASK_ALIASES`
`57`		`-from ..utils.import_utils import is_ipex_version, is_transformers_version`
	`57`	`+from ..utils.import_utils import is_ipex_version, is_torch_version, is_transformers_version`
`58`	`58`	`from ..utils.modeling_utils import recursive_to_device`
`59`	`59`
`60`	`60`