Mirrors
/
EveryDream-trainer
mirror of https://github.com/victorchall/EveryDream-trainer.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

print bad image correctly this time

This commit is contained in:
Victor Hall 2023-01-17 16:42:08 -05:00
parent 4eac671544
commit ed05f3aaa9
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ldm/data/data_loader.py
View File

@ -72,9 +72,9 @@ class DataLoaderMultiAspect():
try: try:
image = Image.open(pathname).convert('RGB') image = Image.open(pathname).convert('RGB')
except Exception as e: except Exception as e:
print(f"File corrupt. Cannot continue. Error opening image: {self.pathname}") print(f"File corrupt. Cannot continue. Error opening image: {pathname}")
print(e) print(e)
print(f"You should remove this image or try to correct it by resaving it.") print(f"You should remove this image or try to correct it by resaving it in an image editor.")
exit() exit()
width, height = image.size width, height = image.size