From 8d0819c548421ca867b7104c53fee25d7b9bff23 Mon Sep 17 00:00:00 2001
From: stduhpf <stephduh@live.fr>
Date: Thu, 3 Jul 2025 16:39:57 +0200
Subject: [PATCH] fix: actually use embeddings with SDXL (#657)

---
 conditioner.hpp | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/conditioner.hpp b/conditioner.hpp
index 4124a68..3f89d52 100644
--- a/conditioner.hpp
+++ b/conditioner.hpp
@@ -459,8 +459,8 @@ struct FrozenCLIPEmbedderWithCustomWords : public Conditioner {
                 if (sd_version_is_sdxl(version)) {
                     text_model2->compute(n_threads,
                                          input_ids2,
-                                         0,
-                                         NULL,
+                                         num_custom_embeddings,
+                                         token_embed_custom.data(),
                                          max_token_idx,
                                          false,
                                          &chunk_hidden_states2, work_ctx);
@@ -470,8 +470,8 @@ struct FrozenCLIPEmbedderWithCustomWords : public Conditioner {
                     if (chunk_idx == 0) {
                         text_model2->compute(n_threads,
                                              input_ids2,
-                                             0,
-                                             NULL,
+                                             num_custom_embeddings,
+                                             token_embed_custom.data(),
                                              max_token_idx,
                                              true,
                                              &pooled,