Skip to content

Commit

Permalink
remove debug print
Browse files Browse the repository at this point in the history
  • Loading branch information
kohya-ss committed Aug 3, 2023
1 parent 9d7619d commit f3be995
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion library/train_util.py
Original file line number Diff line number Diff line change
Expand Up @@ -3791,7 +3791,6 @@ def get_hidden_states_sdxl(

# pool2 = enc_out["text_embeds"]
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
n_size = 1 if max_token_length is None else max_token_length // 75
Expand Down

0 comments on commit f3be995

Please sign in to comment.