support RGBA images

master
AUTOMATIC 2024-03-31 11:48:11 +03:00
parent b6011b2f3b
commit 870584a568
1 changed files with 1 additions and 1 deletions

View File

@ -196,7 +196,7 @@ class ScriptPostprocessingUpscale(scripts_postprocessing.ScriptPostprocessing):
self.model.to(devices.device) 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(): with torch.no_grad():
in_t = process(pp.image).to(devices.device) in_t = process(pp.image).to(devices.device)