From 870584a568f3665d8ac6f72de5c97b60f35c38e9 Mon Sep 17 00:00:00 2001
From: AUTOMATIC <->
Date: Sun, 31 Mar 2024 11:48:11 +0300
Subject: [PATCH] support RGBA images

---
 scripts/postprocessing_pixelization.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/scripts/postprocessing_pixelization.py b/scripts/postprocessing_pixelization.py
index b44fa05..e62c962 100644
--- a/scripts/postprocessing_pixelization.py
+++ b/scripts/postprocessing_pixelization.py
@@ -196,7 +196,7 @@ class ScriptPostprocessingUpscale(scripts_postprocessing.ScriptPostprocessing):
 
         self.model.to(devices.device)
 
-        pp.image = pp.image.resize((pp.image.width * 4 // pixel_size, pp.image.height * 4 // pixel_size))
+        pp.image = pp.image.resize((pp.image.width * 4 // pixel_size, pp.image.height * 4 // pixel_size)).convert('RGB')
 
         with torch.no_grad():
             in_t = process(pp.image).to(devices.device)