This commit is contained in:
w-e-w 2024-12-26 06:26:47 +09:00
parent 6577e063d1
commit f113474a6e
2 changed files with 0 additions and 9 deletions

View File

@ -29,14 +29,10 @@ class ScriptPostprocessingCodeFormer(scripts_postprocessing.ScriptPostprocessing
res = Image.fromarray(restored_img)
if codeformer_visibility < 1.0:
# Ensure consistent size
if pp.image.size != res.size:
res = res.resize(pp.image.size)
# Ensure consistent mode
if pp.image.mode != res.mode:
res = res.convert(pp.image.mode)
res = Image.blend(pp.image, res, codeformer_visibility)
pp.image = res

View File

@ -26,15 +26,10 @@ class ScriptPostprocessingGfpGan(scripts_postprocessing.ScriptPostprocessing):
res = Image.fromarray(restored_img)
if gfpgan_visibility < 1.0:
# Ensure consistent size
if pp.image.size != res.size:
res = res.resize(pp.image.size)
# Ensure consistent mode
if pp.image.mode != res.mode:
res = res.convert(pp.image.mode)
res = Image.blend(pp.image, res, gfpgan_visibility)
pp.image = res