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 sdxl_gen_img not working

Browse Source
This commit is contained in:
Kohya S
2023-08-05 09:32:01 +09:00
parent f4935dd6be
commit 25d8cd473e
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

@@ -981,7 +981,7 @@ def get_unweighted_text_embeddings(
text_embeddings = enc_out["hidden_states"][-2] text_embeddings = enc_out["hidden_states"][-2]
pool = enc_out.get("text_embeds", None) # text encoder 1 doesn't return this pool = enc_out.get("text_embeds", None) # text encoder 1 doesn't return this
if pool is not None: 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 return text_embeddings, pool