diff --git a/model-zoo/models/cxr_image_synthesis_latent_diffusion_model/scripts/saver.py b/model-zoo/models/cxr_image_synthesis_latent_diffusion_model/scripts/saver.py
index b945ceec..05e88722 100644
--- a/model-zoo/models/cxr_image_synthesis_latent_diffusion_model/scripts/saver.py
+++ b/model-zoo/models/cxr_image_synthesis_latent_diffusion_model/scripts/saver.py
@@ -11,7 +11,7 @@ def __init__(self, output_dir: str) -> None:
         self.output_dir = output_dir
 
     def save(self, image_data: torch.Tensor, file_name: str) -> None:
-        image_data = image_data.cpu().numpy()
+        image_data = np.clip(image_data.cpu().numpy(), 0, 1)
         image_data = (image_data * 255).astype(np.uint8)
         im = Image.fromarray(image_data[0, 0])
         im.save(self.output_dir + "/" + file_name + ".jpg")