add the two fixes that @SB2020-eye reported

ZeroCool940711 · Jan 11, 2023 · 19769e1 · 19769e1
1 parent 74296d8
commit 19769e1
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git 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
@@ -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
@@ -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',