From 25d8cd473ec1fdd69510e7d2fe8f38f73c539320 Mon Sep 17 00:00:00 2001 From: Kohya S Date: Sat, 5 Aug 2023 09:32:01 +0900 Subject: [PATCH] fix sdxl_gen_img not working --- sdxl_gen_img.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sdxl_gen_img.py b/sdxl_gen_img.py index 2544c689..68f0e5db 100644 --- a/sdxl_gen_img.py +++ b/sdxl_gen_img.py @@ -981,7 +981,7 @@ def get_unweighted_text_embeddings( text_embeddings = enc_out["hidden_states"][-2] pool = enc_out.get("text_embeds", None) # text encoder 1 doesn't return this if pool is not None: - pool = train_util.pool_workaround(text_encoder, enc_out["last_hidden_state"], text_input_chunk, eos) + pool = train_util.pool_workaround(text_encoder, enc_out["last_hidden_state"], text_input, eos) return text_embeddings, pool