version 1.5.4

minor changes to help with google colab compat

imaginairy/img_log.py

@@ -22,13 +22,7 @@ def log_conditioning(conditioning, description):
 def log_latent(latents, description):
     if _CURRENT_LOGGING_CONTEXT is None:
         return
-    if torch.isnan(latents).any() or torch.isinf(latents).any():
-        logger.error(
-            "Inf/NaN values showing in transformer."
-            + repr(latents)[:50]
-            + " "
-            + description[:50]
-        )
+
     _CURRENT_LOGGING_CONTEXT.log_latents(latents, description)
 
 

setup.py

@@ -40,6 +40,7 @@ setup(
         "tqdm",
         "diffusers",
         "imageio==2.9.0",
+        "Pillow>=8.0.0",
         "pytorch-lightning==1.4.2",
         "omegaconf==2.1.1",
         "einops==0.3.0",