Also add a check for SD dataset

nikita-savelyevv · nikita-savelyevv · commit 9fde5e43fd4b · 2024-08-14T09:23:51.000+02:00
diff --git a/optimum/intel/openvino/quantization.py b/optimum/intel/openvino/quantization.py
@@ -357,6 +357,8 @@ def _quantize_ovbasemodel(
                 if isinstance(self.model, OVModelForCausalLM):
                     calibration_dataset = self._prepare_causal_lm_dataset(quantization_config)
                 elif is_diffusers_available() and isinstance(self.model, OVStableDiffusionPipelineBase):
+                    if not isinstance(quantization_config.dataset, str):
+                        raise ValueError("Please provide dataset as one of the accepted dataset labels.")
                     calibration_dataset = self._prepare_unet_dataset(
                         quantization_config.num_samples, dataset_name=quantization_config.dataset
                     )

Original file line number	Diff line number	Diff line change
`@@ -357,6 +357,8 @@ def _quantize_ovbasemodel(`
`357`	`357`	`if isinstance(self.model, OVModelForCausalLM):`
`358`	`358`	`calibration_dataset = self._prepare_causal_lm_dataset(quantization_config)`
`359`	`359`	`elif is_diffusers_available() and isinstance(self.model, OVStableDiffusionPipelineBase):`
	`360`	`+ if not isinstance(quantization_config.dataset, str):`
	`361`	`+ raise ValueError("Please provide dataset as one of the accepted dataset labels.")`
`360`	`362`	`calibration_dataset = self._prepare_unet_dataset(`
`361`	`363`	`quantization_config.num_samples, dataset_name=quantization_config.dataset`
`362`	`364`	`)`