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

Revert

This commit is contained in:
Maw-Fox 2022-11-10 06:11:09 -07:00
parent 0be39a4887
commit 46e8d98d2b
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
diffusers_trainer.py
View File

@ -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)