print削除

This commit is contained in:
u-haru
2023-03-26 02:26:08 +09:00
parent 4dc1124f93
commit 5a3d564a30

View File

@@ -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):