We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent e394866 commit 4f7e87cCopy full SHA for 4f7e87c
optimum/intel/openvino/modeling_diffusion.py
@@ -312,7 +312,7 @@ def _from_pretrained(
312
if not isinstance(sd_model, supported_pipelines):
313
raise NotImplementedError(f"Quantization in hybrid mode is not supported for {cls.__name__}")
314
315
- num_inference_steps = 4 if isinstance(cls, OVLatentConsistencyModelPipeline) else 50
+ num_inference_steps = 4 if isinstance(sd_model, OVLatentConsistencyModelPipeline) else 50
316
quantization_config.dataset = dataset
317
318
if isinstance(quantization_config.dataset, str):
0 commit comments