From 066edad3191ff4d08b7c875a044b5a231db170f0 Mon Sep 17 00:00:00 2001
From: Nourepide <nourepide@gmail.com>
Date: Fri, 24 Nov 2023 14:52:37 +0300
Subject: [PATCH] Fix ImageTransformCropCorners with improved alpha channel
 handling and antialiasing

---
 modules/ImageTransform.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/modules/ImageTransform.py b/modules/ImageTransform.py
index d7d34bb..7376303 100644
--- a/modules/ImageTransform.py
+++ b/modules/ImageTransform.py
@@ -231,7 +231,7 @@ class ImageTransformCropCorners:
 
         def crop_tensor(tensor):
             return torch.stack([
-                tensor[:, :, i] - mask for i in range(tensor.shape[2])
+                (tensor[:, :, i] - mask).clamp(0, 1) for i in range(tensor.shape[2])
             ], dim=2)
 
         return (torch.stack([