diff --git a/library/train_util.py b/library/train_util.py index 994201fc..1ba38adc 100644 --- a/library/train_util.py +++ b/library/train_util.py @@ -517,7 +517,6 @@ class BaseDataset(torch.utils.data.Dataset): else: caption = caption.replace(str_from, str_to) - print(self.current_step, self.max_train_steps, caption) return caption def get_input_ids(self, caption):