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

remove debug print

Browse Source
This commit is contained in:
Kohya S
2023-08-04 08:44:17 +09:00
parent 9d7619d1eb
commit f3be995c28
1 changed files with 0 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
library/train_util.py
View File

@@ -3791,7 +3791,6 @@ def get_hidden_states_sdxl(
# pool2 = enc_out["text_embeds"] # pool2 = enc_out["text_embeds"]
pool2 = pool_workaround(text_encoder2, enc_out["last_hidden_state"], input_ids2, tokenizer2.eos_token_id) pool2 = pool_workaround(text_encoder2, enc_out["last_hidden_state"], input_ids2, tokenizer2.eos_token_id)
print(f"original pool2: {enc_out['text_embeds']}, fixed: {pool2}")
# b*n, 77, 768 or 1280 -> b, n*77, 768 or 1280 # b*n, 77, 768 or 1280 -> b, n*77, 768 or 1280
n_size = 1 if max_token_length is None else max_token_length // 75 n_size = 1 if max_token_length is None else max_token_length // 75