support RGBA images

2024-03-31 11:48:11 +03:00 · 2024-03-31 11:48:11 +03:00 · 870584a568
parent b6011b2f3b
commit 870584a568
1 changed files with 1 additions and 1 deletions
--- 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)