diff --git a/run.py b/run.py
index 193491b..a29e3ff 100644
--- a/run.py
+++ b/run.py
@@ -184,8 +184,8 @@ def cpu_call(self, original_image: np.ndarray):
             with torch.autocast(device_type=self.cfg.MODEL.DEVICE, enabled=self.cfg.MODEL.AUTOCAST):
                 predictions = self.model([inputs])[0]
 
-            if torch.isnan(predictions["sem_seg"]).any():
-                raise ValueError("NaN in predictions")
+            # if torch.isnan(predictions["sem_seg"]).any():
+            #     raise ValueError("NaN in predictions")
 
         return predictions, height, width