Skip to content

Commit 8d0819c

Browse files
authored
fix: actually use embeddings with SDXL (#657)
1 parent 7a8ff2e commit 8d0819c

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

conditioner.hpp

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -459,8 +459,8 @@ struct FrozenCLIPEmbedderWithCustomWords : public Conditioner {
459459
if (sd_version_is_sdxl(version)) {
460460
text_model2->compute(n_threads,
461461
input_ids2,
462-
0,
463-
NULL,
462+
num_custom_embeddings,
463+
token_embed_custom.data(),
464464
max_token_idx,
465465
false,
466466
&chunk_hidden_states2, work_ctx);
@@ -470,8 +470,8 @@ struct FrozenCLIPEmbedderWithCustomWords : public Conditioner {
470470
if (chunk_idx == 0) {
471471
text_model2->compute(n_threads,
472472
input_ids2,
473-
0,
474-
NULL,
473+
num_custom_embeddings,
474+
token_embed_custom.data(),
475475
max_token_idx,
476476
true,
477477
&pooled,

0 commit comments

Comments
 (0)