Fix transformers version v4.38 compatibility (#657)

Author: echarlaix

optimum/intel/neural_compressor/trainer.py

@@ -62,7 +62,6 @@
     is_accelerate_available,
     is_apex_available,
     is_sagemaker_mp_enabled,
-    is_torch_xla_available,
     logging,
 )
 
@@ -73,6 +72,12 @@
 from .configuration import INCConfig
 
 
+if is_transformers_version(">=", "4.39.0"):
+    from transformers.utils import is_torch_xla_available
+else:
+    from transformers.utils import is_torch_tpu_available as is_torch_xla_available
+
+
 if is_accelerate_available():
     from accelerate import __version__ as accelerate_version
     from accelerate import skip_first_batches

optimum/intel/openvino/trainer.py

@@ -80,7 +80,6 @@
     is_accelerate_available,
     is_apex_available,
     is_sagemaker_mp_enabled,
-    is_torch_xla_available,
     logging,
 )
 
@@ -101,6 +100,11 @@
 )
 
 
+if is_transformers_version(">=", "4.39.0"):
+    from transformers.utils import is_torch_xla_available
+else:
+    from transformers.utils import is_torch_tpu_available as is_torch_xla_available
+
 if is_accelerate_available():
     from accelerate import __version__ as accelerate_version
     from accelerate import skip_first_batches