Skip to content

Commit 4f7e87c

Browse files
committed
fix lcm bug
1 parent e394866 commit 4f7e87c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

optimum/intel/openvino/modeling_diffusion.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -312,7 +312,7 @@ def _from_pretrained(
312312
if not isinstance(sd_model, supported_pipelines):
313313
raise NotImplementedError(f"Quantization in hybrid mode is not supported for {cls.__name__}")
314314

315-
num_inference_steps = 4 if isinstance(cls, OVLatentConsistencyModelPipeline) else 50
315+
num_inference_steps = 4 if isinstance(sd_model, OVLatentConsistencyModelPipeline) else 50
316316
quantization_config.dataset = dataset
317317

318318
if isinstance(quantization_config.dataset, str):

0 commit comments

Comments
 (0)