From 46e8d98d2b5fd323a28b73756964f7b634b91af0 Mon Sep 17 00:00:00 2001 From: Maw-Fox Date: Thu, 10 Nov 2022 06:11:09 -0700 Subject: [PATCH] Revert --- diffusers_trainer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/diffusers_trainer.py b/diffusers_trainer.py index c5d0bd7..00c076d 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(Image.LANCZOS) + method=Image.Resampling.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)