Skip to content

Commit 0a8fba0

Browse files
Fix
1 parent c566ccc commit 0a8fba0

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

optimum/commands/export/openvino.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -203,7 +203,7 @@ def parse_args(parser: "ArgumentParser"):
203203
return parse_args_openvino(parser)
204204

205205
def run(self):
206-
from ...exporters.openvino.__main__ import main_export, infer_task
206+
from ...exporters.openvino.__main__ import infer_task, main_export
207207
from ...intel.openvino.configuration import _DEFAULT_4BIT_CONFIGS, OVConfig
208208

209209
if self.args.fp16:
@@ -321,10 +321,10 @@ def run(self):
321321
export_tokenizer(tokenizer_2, output / "tokenizer_2")
322322
else:
323323
task = infer_task(self.args.task, self.args.model)
324-
quantization_config = ov_config.quantization_config
324+
quantization_config = ov_config.quantization_config if ov_config else None
325325
quantize_after_export = (
326326
task.startswith("text-generation")
327-
and quantization_config is not None
327+
and quantization_config
328328
and hasattr(quantization_config, "dataset")
329329
and quantization_config.dataset is not None
330330
)

0 commit comments

Comments
 (0)