From 7ae0cde75464c44e29b1bbd1983078d2587d9892 Mon Sep 17 00:00:00 2001
From: Kohya S <ykumeykume@gmail.com>
Date: Sun, 23 Jul 2023 15:18:27 +0900
Subject: [PATCH] fix max mul embeds doesn't work. closes #656

---
 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 b5e26105..d2f59a33 100644
--- a/sdxl_gen_img.py
+++ b/sdxl_gen_img.py
@@ -958,7 +958,7 @@ def get_unweighted_text_embeddings(
             enc_out = text_encoder(text_input_chunk, output_hidden_states=True, return_dict=True)
             text_embedding = enc_out["hidden_states"][-2]
             if pool is None:
-                pool = enc_out["text_embeds"]  # use 1st chunk
+                pool = enc_out.get("text_embeds", None)  # use 1st chunk, if provided
 
             if no_boseos_middle:
                 if i == 0: