VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-08 22:35:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix max mul embeds doesn't work. closes #656

Browse Source
This commit is contained in:
Kohya S
2023-07-23 15:18:27 +09:00
parent c1d5c24bc7
commit 7ae0cde754
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
sdxl_gen_img.py
View File

@@ -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: