diff --git a/library/train_util.py b/library/train_util.py index 5e152532..8ea7a438 100644 --- a/library/train_util.py +++ b/library/train_util.py @@ -713,7 +713,6 @@ class BaseDataset(torch.utils.data.Dataset): else: caption = caption.replace(str_from, str_to) - print(caption) return caption def get_input_ids(self, caption, tokenizer=None):