Skip to content
This repository was archived by the owner on Nov 27, 2024. It is now read-only.

Commit 6626eb5

Browse files
committed
Fix issues found during testing
1 parent d54393e commit 6626eb5

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

OnnxStack.StableDiffusion/Pipelines/StableDiffusionXLPipeline.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -293,7 +293,7 @@ private async Task<PromptEmbeddingsResult> GenerateEmbedsAsync(TokenizerResult i
293293
}
294294

295295
var promptTensor = new DenseTensor<float>(promptEmbeddings.ToArray(), new[] { 1, promptEmbeddings.Count / _tokenizer2.TokenizerLength, _tokenizer2.TokenizerLength });
296-
var pooledTensor = new DenseTensor<float>(pooledPromptEmbeddings.ToArray(), new[] { 1, pooledPromptEmbeddings.Count });
296+
var pooledTensor = new DenseTensor<float>(pooledPromptEmbeddings.Take(_tokenizer2.TokenizerLength).ToArray(), new[] { 1, _tokenizer2.TokenizerLength });
297297
return new PromptEmbeddingsResult(promptTensor, pooledTensor);
298298
}
299299

0 commit comments

Comments
 (0)