From 0be39a4887918e9cba1cf02c6ec66499768e62d2 Mon Sep 17 00:00:00 2001 From: Maw-Fox Date: Thu, 10 Nov 2022 05:42:38 -0700 Subject: [PATCH] Fix depreciated enum --- diffusers_trainer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/diffusers_trainer.py b/diffusers_trainer.py index 00c076d..c5d0bd7 100644 --- 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) @@ -548,7 +548,7 @@ def main(): if args.resume: args.model = args.resume - + tokenizer = CLIPTokenizer.from_pretrained(args.model, subfolder='tokenizer', use_auth_token=args.hf_token) text_encoder = CLIPTextModel.from_pretrained(args.model, subfolder='text_encoder', use_auth_token=args.hf_token) vae = AutoencoderKL.from_pretrained(args.model, subfolder='vae', use_auth_token=args.hf_token)