fix: actually use embeddings with SDXL (#657)

This commit is contained in:
stduhpf 2025-07-03 16:39:57 +02:00 committed by GitHub
parent 7a8ff2e819
commit 8d0819c548
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

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,