diff --git a/scripts/postprocessing_codeformer.py b/scripts/postprocessing_codeformer.py index f86e99a04..d00858918 100644 --- a/scripts/postprocessing_codeformer.py +++ b/scripts/postprocessing_codeformer.py @@ -29,6 +29,10 @@ class ScriptPostprocessingCodeFormer(scripts_postprocessing.ScriptPostprocessing res = Image.fromarray(restored_img) if codeformer_visibility < 1.0: + if pp.image.size != res.size: + res = res.resize(pp.image.size) + if pp.image.mode != res.mode: + res = res.convert(pp.image.mode) res = Image.blend(pp.image, res, codeformer_visibility) pp.image = res diff --git a/scripts/postprocessing_gfpgan.py b/scripts/postprocessing_gfpgan.py index 3a130fd63..574817a92 100644 --- a/scripts/postprocessing_gfpgan.py +++ b/scripts/postprocessing_gfpgan.py @@ -26,6 +26,10 @@ class ScriptPostprocessingGfpGan(scripts_postprocessing.ScriptPostprocessing): res = Image.fromarray(restored_img) if gfpgan_visibility < 1.0: + if pp.image.size != res.size: + res = res.resize(pp.image.size) + if pp.image.mode != res.mode: + res = res.convert(pp.image.mode) res = Image.blend(pp.image, res, gfpgan_visibility) pp.image = res