Fix depreciated enum

2022-11-10 05:42:38 -07:00 · 2022-11-10 05:42:38 -07:00 · 0be39a4887
parent 3fe9df1450
commit 0be39a4887
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.LANCZOS
+                method=Image.Resampling(Image.LANCZOS)
            )

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