diff --git a/OnnxStack.StableDiffusion/Diffusers/LatentConsistency/LatentConsistencyDiffuser.cs b/OnnxStack.StableDiffusion/Diffusers/LatentConsistency/LatentConsistencyDiffuser.cs index b365d5ce..1d1a5819 100644 --- a/OnnxStack.StableDiffusion/Diffusers/LatentConsistency/LatentConsistencyDiffuser.cs +++ b/OnnxStack.StableDiffusion/Diffusers/LatentConsistency/LatentConsistencyDiffuser.cs @@ -108,9 +108,6 @@ protected override async Task> SchedulerStepAsync(StableDiffu // Get Model metadata var metadata = _onnxModelService.GetModelMetadata(modelOptions, OnnxModelType.Unet); - // Some LCM variants require no guidance embeds - var guidanceEmbeddingsRequired = metadata.Inputs.Count == 4; - // Loop though the timesteps var step = 0; foreach (var timestep in timesteps) @@ -130,8 +127,7 @@ protected override async Task> SchedulerStepAsync(StableDiffu inferenceParameters.AddInputTensor(inputTensor); inferenceParameters.AddInputTensor(timestepTensor); inferenceParameters.AddInputTensor(promptEmbeddings.PromptEmbeds); - if(guidanceEmbeddingsRequired) - inferenceParameters.AddInputTensor(guidanceEmbeddings); + inferenceParameters.AddInputTensor(guidanceEmbeddings); inferenceParameters.AddOutputBuffer(outputDimension); var results = await _onnxModelService.RunInferenceAsync(modelOptions, OnnxModelType.Unet, inferenceParameters);