From 32457320ead060a65bbd55e1bd53f38ffe9c6e19 Mon Sep 17 00:00:00 2001 From: Ella Charlaix Date: Mon, 6 Nov 2023 13:09:35 +0100 Subject: [PATCH] format --- optimum/intel/neural_compressor/trainer.py | 3 --- optimum/intel/openvino/trainer.py | 3 --- 2 files changed, 6 deletions(-) diff --git a/optimum/intel/neural_compressor/trainer.py b/optimum/intel/neural_compressor/trainer.py index c648f10432..918a2e4885 100644 --- a/optimum/intel/neural_compressor/trainer.py +++ b/optimum/intel/neural_compressor/trainer.py @@ -43,9 +43,6 @@ from transformers import Trainer from transformers.data.data_collator import DataCollator from transformers.debug_utils import DebugOption, DebugUnderflowOverflow - -# Integrations must be imported before ML frameworks: -from transformers.integrations import deepspeed_init, deepspeed_load_checkpoint, hp_params, is_deepspeed_available from transformers.modeling_utils import PreTrainedModel, get_parameter_dtype, unwrap_model from transformers.models.auto.modeling_auto import MODEL_FOR_CAUSAL_LM_MAPPING_NAMES from transformers.pytorch_utils import is_torch_less_than_1_11 diff --git a/optimum/intel/openvino/trainer.py b/optimum/intel/openvino/trainer.py index 7bbfc4599e..dfc659882c 100644 --- a/optimum/intel/openvino/trainer.py +++ b/optimum/intel/openvino/trainer.py @@ -62,9 +62,6 @@ from transformers import Trainer from transformers.data.data_collator import DataCollator from transformers.debug_utils import DebugOption, DebugUnderflowOverflow - -# Integrations must be imported before ML frameworks: -from transformers.integrations import deepspeed_init, deepspeed_load_checkpoint, hp_params, is_deepspeed_available from transformers.modeling_utils import PreTrainedModel, unwrap_model from transformers.pytorch_utils import is_torch_less_than_1_11 from transformers.tokenization_utils_base import PreTrainedTokenizerBase