Skip to content

Commit a42de51

Browse files
committed
fix tests
1 parent 6f28328 commit a42de51

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

optimum/commands/export/openvino.py

+6-1
Original file line numberDiff line numberDiff line change
@@ -211,7 +211,12 @@ def run(self):
211211

212212
library_name = TasksManager.infer_library_from_model(self.args.model)
213213

214-
if library_name == "diffusers" and ov_config and ov_config.quantization_config.get("dataset"):
214+
if (
215+
library_name == "diffusers"
216+
and ov_config
217+
and ov_config.quantization_config
218+
and ov_config.quantization_config.dataset is not None
219+
):
215220
if not is_diffusers_available():
216221
raise ValueError(DIFFUSERS_IMPORT_ERROR.format("Export of diffusers models"))
217222

0 commit comments

Comments
 (0)