From f3be995c281f3385945510bc35608cf52c2f6f8e Mon Sep 17 00:00:00 2001 From: Kohya S Date: Fri, 4 Aug 2023 08:44:17 +0900 Subject: [PATCH] remove debug print --- library/train_util.py | 1 - 1 file changed, 1 deletion(-) diff --git a/library/train_util.py b/library/train_util.py index 3eda5098..23f7f634 100644 --- a/library/train_util.py +++ b/library/train_util.py @@ -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