From 19769e17c9612847d6119b7822892c69ec1e0e7f Mon Sep 17 00:00:00 2001 From: Phil Wang Date: Wed, 11 Jan 2023 10:02:14 -0800 Subject: [PATCH] add the two fixes that @SB2020-eye reported --- muse_maskgit_pytorch/t5.py | 1 + muse_maskgit_pytorch/trainers.py | 3 ++- setup.py | 2 +- 3 files changed, 4 insertions(+), 2 deletions(-) diff --git a/muse_maskgit_pytorch/t5.py b/muse_maskgit_pytorch/t5.py index 78ff0a5..6833f71 100644 --- a/muse_maskgit_pytorch/t5.py +++ b/muse_maskgit_pytorch/t5.py @@ -1,3 +1,4 @@ +import logging import torch import transformers from transformers import T5Tokenizer, T5EncoderModel, T5Config diff --git a/muse_maskgit_pytorch/trainers.py b/muse_maskgit_pytorch/trainers.py index 3c1bcd7..a736f2a 100644 --- a/muse_maskgit_pytorch/trainers.py +++ b/muse_maskgit_pytorch/trainers.py @@ -22,7 +22,8 @@ from ema_pytorch import EMA -from PIL import Image +from PIL import Image, ImageFile +ImageFile.LOAD_TRUNCATED_IMAGES = True # helper functions diff --git a/setup.py b/setup.py index dc90fdb..ebbf31c 100644 --- a/setup.py +++ b/setup.py @@ -3,7 +3,7 @@ setup( name = 'muse-maskgit-pytorch', packages = find_packages(exclude=[]), - version = '0.0.16', + version = '0.0.17', license='MIT', description = 'MUSE - Text-to-Image Generation via Masked Generative Transformers, in Pytorch', author = 'Phil Wang',