Mirrors
/
waifu-diffusion
mirror of https://github.com/harubaru/waifu-diffusion.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix deprecation warnings in dataloader

This commit is contained in:
harubaru 2022-10-15 18:43:28 -07:00
parent 67140231e7
commit 76efa00de3
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
ldm/data/localdanboorubase.py
View File

@ -140,10 +140,10 @@ class LocalDanbooruBase(Dataset):
print(f'image-caption map has {len(self.examples.keys())} examples') print(f'image-caption map has {len(self.examples.keys())} examples')
self.size = size self.size = size
self.interpolation = {"linear": PIL.Image.LINEAR, self.interpolation = {"linear": PIL.Image.Resampling.BILINEAR,
"bilinear": PIL.Image.BILINEAR, "bilinear": PIL.Image.Resampling.BILINEAR,
"bicubic": PIL.Image.BICUBIC, "bicubic": PIL.Image.Resampling.BICUBIC,
"lanczos": PIL.Image.LANCZOS, "lanczos": PIL.Image.Resampling.LANCZOS,
}[interpolation] }[interpolation]
self.flip = transforms.RandomHorizontalFlip(p=flip_p) self.flip = transforms.RandomHorizontalFlip(p=flip_p)