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

Merge pull request #716 from kohya-ss/dev

Browse Source 
fix sdxl_gen_img not working
This commit is contained in:
Kohya S
2023-08-05 09:33:19 +09:00
committed by GitHub
parent f4935dd6be 25d8cd473e
commit 2dfa26cca0
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]
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