Skip to content

Commit fbbe804

Browse files
committed
style
1 parent bae24bf commit fbbe804

File tree

3 files changed

+2
-5
lines changed

3 files changed

+2
-5
lines changed

optimum/intel/openvino/modeling_base.py

+1-3
Original file line numberDiff line numberDiff line change
@@ -152,12 +152,10 @@ def _save_pretrained(self, save_directory: Union[str, Path]):
152152

153153
def _save_openvino_config(self, save_directory: Union[str, Path]):
154154
if self._openvino_config is not None:
155-
156155
if not isinstance(self._openvino_config.quantization_config.dataset, (str, type(None))):
157156
self._openvino_config.quantization_config.dataset = None
158-
159-
self._openvino_config.save_pretrained(save_directory)
160157

158+
self._openvino_config.save_pretrained(save_directory)
161159

162160
@classmethod
163161
def _from_pretrained(

optimum/intel/openvino/modeling_decoder.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -12,9 +12,9 @@
1212
# See the License for the specific language governing permissions and
1313
# limitations under the License.
1414

15+
import copy
1516
import logging
1617
import os
17-
import copy
1818
from pathlib import Path
1919
from tempfile import TemporaryDirectory
2020
from typing import Dict, Optional, Tuple, Union

tests/openvino/test_quantization.py

-1
Original file line numberDiff line numberDiff line change
@@ -109,7 +109,6 @@ def preprocess_function(examples, tokenizer):
109109
loaded_config = OVConfig.from_pretrained(tmp_dir)
110110
self.assertEqual(DEFAULT_QUANTIZATION_CONFIG, loaded_config.to_dict()["compression"])
111111

112-
113112
@parameterized.expand(SUPPORTED_ARCHITECTURES_WITH_EXPECTED_QUANTIZED_MATMULS)
114113
def test_ovmodel_static_quantization(self, model_cls, model_name, expected_fake_quantize, expected_int8):
115114
task = model_cls.export_feature

0 commit comments

Comments
 (0)