mirror of
https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
synced 2025-05-06 03:49:02 +08:00
lint
This commit is contained in:
parent
6577e063d1
commit
f113474a6e
@ -29,14 +29,10 @@ class ScriptPostprocessingCodeFormer(scripts_postprocessing.ScriptPostprocessing
|
|||||||
res = Image.fromarray(restored_img)
|
res = Image.fromarray(restored_img)
|
||||||
|
|
||||||
if codeformer_visibility < 1.0:
|
if codeformer_visibility < 1.0:
|
||||||
# Ensure consistent size
|
|
||||||
if pp.image.size != res.size:
|
if pp.image.size != res.size:
|
||||||
res = res.resize(pp.image.size)
|
res = res.resize(pp.image.size)
|
||||||
|
|
||||||
# Ensure consistent mode
|
|
||||||
if pp.image.mode != res.mode:
|
if pp.image.mode != res.mode:
|
||||||
res = res.convert(pp.image.mode)
|
res = res.convert(pp.image.mode)
|
||||||
|
|
||||||
res = Image.blend(pp.image, res, codeformer_visibility)
|
res = Image.blend(pp.image, res, codeformer_visibility)
|
||||||
|
|
||||||
pp.image = res
|
pp.image = res
|
||||||
|
@ -26,15 +26,10 @@ class ScriptPostprocessingGfpGan(scripts_postprocessing.ScriptPostprocessing):
|
|||||||
res = Image.fromarray(restored_img)
|
res = Image.fromarray(restored_img)
|
||||||
|
|
||||||
if gfpgan_visibility < 1.0:
|
if gfpgan_visibility < 1.0:
|
||||||
|
|
||||||
# Ensure consistent size
|
|
||||||
if pp.image.size != res.size:
|
if pp.image.size != res.size:
|
||||||
res = res.resize(pp.image.size)
|
res = res.resize(pp.image.size)
|
||||||
|
|
||||||
# Ensure consistent mode
|
|
||||||
if pp.image.mode != res.mode:
|
if pp.image.mode != res.mode:
|
||||||
res = res.convert(pp.image.mode)
|
res = res.convert(pp.image.mode)
|
||||||
|
|
||||||
res = Image.blend(pp.image, res, gfpgan_visibility)
|
res = Image.blend(pp.image, res, gfpgan_visibility)
|
||||||
|
|
||||||
pp.image = res
|
pp.image = res
|
||||||
|
Loading…
x
Reference in New Issue
Block a user