Update train_util.py

I removed the sleep since it triggers per subset and if someone had a lot of subsets it would trigger multiple times
This commit is contained in:
TingTingin
2023-05-26 08:08:34 -04:00
committed by GitHub
parent 061e157191
commit db756e9a34

View File

@@ -1109,7 +1109,6 @@ class DreamBoothDataset(BaseDataset):
print(missing_caption+f"... and {remaining_missing_captions} more") print(missing_caption+f"... and {remaining_missing_captions} more")
break break
print(missing_caption) print(missing_caption)
time.sleep(5)
return img_paths, captions return img_paths, captions
print("prepare images.") print("prepare images.")