From 420d453dff24985e194638f5cc753d3fc5486159 Mon Sep 17 00:00:00 2001 From: Nikita Savelyev Date: Mon, 22 Apr 2024 17:17:55 +0200 Subject: [PATCH] Move import --- optimum/intel/openvino/quantization.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/optimum/intel/openvino/quantization.py b/optimum/intel/openvino/quantization.py index 07765920bc..89d620d45c 100644 --- a/optimum/intel/openvino/quantization.py +++ b/optimum/intel/openvino/quantization.py @@ -21,7 +21,6 @@ from typing import Any, Callable, Dict, Iterable, List, Optional, Tuple, Union import datasets -from transformers.utils.quantization_config import QuantizationMethod import nncf import openvino @@ -38,6 +37,7 @@ from transformers import AutoTokenizer, DataCollator, PreTrainedModel, default_data_collator from transformers.pytorch_utils import Conv1D from transformers.utils import is_accelerate_available +from transformers.utils.quantization_config import QuantizationMethod from optimum.exporters.onnx.convert import check_dummy_inputs_are_allowed from optimum.exporters.tasks import TasksManager