Revert

2022-11-10 06:11:09 -07:00 · 2022-11-10 06:11:09 -07:00 · 46e8d98d2b
parent 0be39a4887
commit 46e8d98d2b
1 changed files with 2 additions and 2 deletions
--- a/diffusers_trainer.py
+++ b/diffusers_trainer.py
@ -401,7 +401,7 @@ class AspectDataset(torch.utils.data.Dataset):
                (item[1], item[2]),
                bleed=0.0,
                centering=(0.5, 0.5),
-                method=Image.Resampling(Image.LANCZOS)
+                method=Image.Resampling.LANCZOS
            )

        return_dict['pixel_values'] = self.transforms(image_file)