fix: actually use embeddings with SDXL (#657)

This commit is contained in:
stduhpf authored and GitHub committed 2025-07-03 22:39:57 +08:00
1 parent 7a8ff2e819
commit 8d0819c548
1 file changed
+4 -4
+4 -4
View File
@@ -459,8 +459,8 @@ struct FrozenCLIPEmbedderWithCustomWords : public Conditioner {
if (sd_version_is_sdxl(version)) { if (sd_version_is_sdxl(version)) {
text_model2->compute(n_threads, text_model2->compute(n_threads,
input_ids2, input_ids2,
0, num_custom_embeddings,
NULL, token_embed_custom.data(),
max_token_idx, max_token_idx,
false, false,
&chunk_hidden_states2, work_ctx); &chunk_hidden_states2, work_ctx);
@@ -470,8 +470,8 @@ struct FrozenCLIPEmbedderWithCustomWords : public Conditioner {
if (chunk_idx == 0) { if (chunk_idx == 0) {
text_model2->compute(n_threads, text_model2->compute(n_threads,
input_ids2, input_ids2,
0, num_custom_embeddings,
NULL, token_embed_custom.data(),
max_token_idx, max_token_idx,
true, true,
&pooled, &pooled,