diff --git a/modules/postprocess/pixelart.py b/modules/postprocess/pixelart.py index c1c021a49..d9ce620ee 100644 --- a/modules/postprocess/pixelart.py +++ b/modules/postprocess/pixelart.py @@ -92,7 +92,7 @@ def rgb_to_ycbcr_tensor(image: torch.ByteTensor) -> torch.FloatTensor: @devices.inference_context() def ycbcr_tensor_to_rgb(ycbcr: torch.FloatTensor) -> torch.ByteTensor: - ycbcr_img = (ycbcr / 2) + ycbcr_img = ycbcr / 2 y = ycbcr_img[:,0,:,:].add_(0.5) cb = ycbcr_img[:,1,:,:] cr = ycbcr_img[:,2,:,:]