Skip to content

Commit 1a773bf

Browse files
authored
Merge pull request AUTOMATIC1111#16751 from Neokmi/master
Fix Codeformer and gfpgan extension , Inconsistent overlay layer types when visibility value is less than 1
2 parents f25c3fc + f113474 commit 1a773bf

File tree

2 files changed

+8
-0
lines changed

2 files changed

+8
-0
lines changed

scripts/postprocessing_codeformer.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,10 @@ def process(self, pp: scripts_postprocessing.PostprocessedImage, enable, codefor
2929
res = Image.fromarray(restored_img)
3030

3131
if codeformer_visibility < 1.0:
32+
if pp.image.size != res.size:
33+
res = res.resize(pp.image.size)
34+
if pp.image.mode != res.mode:
35+
res = res.convert(pp.image.mode)
3236
res = Image.blend(pp.image, res, codeformer_visibility)
3337

3438
pp.image = res

scripts/postprocessing_gfpgan.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,10 @@ def process(self, pp: scripts_postprocessing.PostprocessedImage, enable, gfpgan_
2626
res = Image.fromarray(restored_img)
2727

2828
if gfpgan_visibility < 1.0:
29+
if pp.image.size != res.size:
30+
res = res.resize(pp.image.size)
31+
if pp.image.mode != res.mode:
32+
res = res.convert(pp.image.mode)
2933
res = Image.blend(pp.image, res, gfpgan_visibility)
3034

3135
pp.image = res

0 commit comments

Comments
 (0)