We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 6f28328 commit a42de51Copy full SHA for a42de51
optimum/commands/export/openvino.py
@@ -211,7 +211,12 @@ def run(self):
211
212
library_name = TasksManager.infer_library_from_model(self.args.model)
213
214
- if library_name == "diffusers" and ov_config and ov_config.quantization_config.get("dataset"):
+ 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
+ ):
220
if not is_diffusers_available():
221
raise ValueError(DIFFUSERS_IMPORT_ERROR.format("Export of diffusers models"))
222
0 commit comments