ToDType before ToPILImage for uint8 rounding

ToPILImage converts to uint8 without rounding:
(npimg * 255).astype(np.uint8)

ToDtype converts to uint8 while rounding:
image.mul(255 + 1.0 - 0.001).to(torch.uint8)

modules/upscaler_utils.py

@@ -21,7 +21,8 @@ def upscale_pil_patch(model, img: Image.Image) -> Image.Image:
         tensor = T.PILToTensor()(img)
         tensor = T.ToDtype(torch.float32, scale=True)(tensor)
         tensor = tensor.clamp_(0.0, 1.0).unsqueeze(0).to(device=param.device, dtype=param.dtype)
-        return T.ToPILImage(mode="RGB")(model(tensor).squeeze(0).clamp_(0.0, 1.0))
+        tensorimg = T.ToDtype(torch.uint8, scale=True)(model(tensor).squeeze(0).clamp_(0.0, 1.0))
+        return T.ToPILImage(mode="RGB")(tensorimg)
 
 
 def upscale_with_model(
@@ -165,5 +166,6 @@ def upscale_2(
             desc=desc,
             device=param.device,
         )
-    return T.ToPILImage(mode="RGB")(output.squeeze(0).clamp_(0.0, 1.0))
+    output = T.ToDtype(torch.uint8, scale=True)(output.squeeze(0).clamp_(0.0, 1.0))
+    return T.ToPILImage(mode="RGB")(output)