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: