diff --git a/imaginairy/img_log.py b/imaginairy/img_log.py
index 275a46e..56294e0 100644
--- a/imaginairy/img_log.py
+++ b/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)
 
 
diff --git a/setup.py b/setup.py
index 11baf22..c35095b 100644
--- a/setup.py
+++ b/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",