This repository was archived by the owner on Nov 27, 2024. It is now read-only.
File tree Expand file tree Collapse file tree 1 file changed +6
-2
lines changed
OnnxStack.StableDiffusion/Diffusers/LatentConsistency Expand file tree Collapse file tree 1 file changed +6
-2
lines changed Original file line number Diff line number Diff line change @@ -107,7 +107,10 @@ protected override async Task<DenseTensor<float>> SchedulerStepAsync(IModelOptio
107
107
108
108
// Get Model metadata
109
109
var metadata = _onnxModelService . GetModelMetadata ( modelOptions , OnnxModelType . Unet ) ;
110
-
110
+
111
+ // Some LCM variants require no guidance embeds
112
+ var guidanceEmbeddingsRequired = metadata . Inputs . Count == 4 ;
113
+
111
114
// Loop though the timesteps
112
115
var step = 0 ;
113
116
foreach ( var timestep in timesteps )
@@ -127,7 +130,8 @@ protected override async Task<DenseTensor<float>> SchedulerStepAsync(IModelOptio
127
130
inferenceParameters . AddInputTensor ( inputTensor ) ;
128
131
inferenceParameters . AddInputTensor ( timestepTensor ) ;
129
132
inferenceParameters . AddInputTensor ( promptEmbeddings . PromptEmbeds ) ;
130
- inferenceParameters . AddInputTensor ( guidanceEmbeddings ) ;
133
+ if ( guidanceEmbeddingsRequired )
134
+ inferenceParameters . AddInputTensor ( guidanceEmbeddings ) ;
131
135
inferenceParameters . AddOutputBuffer ( outputDimension ) ;
132
136
133
137
var results = await _onnxModelService . RunInferenceAsync ( modelOptions , OnnxModelType . Unet , inferenceParameters ) ;
You can’t perform that action at this time.
0 commit comments