From b1d1063a25fc504bde00bcdd5d4e8f0cd63c957c Mon Sep 17 00:00:00 2001 From: blessedcoolant <54517381+blessedcoolant@users.noreply.github.com> Date: Sat, 24 Sep 2022 19:54:22 +1200 Subject: [PATCH] Fix CodeFormer not working if GFPGAN is enabeld --- ldm/generate.py | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) diff --git a/ldm/generate.py b/ldm/generate.py index 9471d01f87..11ac655793 100644 --- a/ldm/generate.py +++ b/ldm/generate.py @@ -727,16 +727,16 @@ class Generate: print(">> ESRGAN is disabled. Image not upscaled.") if strength > 0: if self.gfpgan is not None or self.codeformer is not None: - if self.gfpgan is None: - if facetool == 'codeformer': - if self.codeformer is not None: - image = self.codeformer.process(image=image, strength=strength, device=self.device, seed=seed, fidelity=codeformer_fidelity) - else: - print('>> CodeFormer not found. Face restoration is disabled.') - else: + if facetool == 'gfpgan': + if self.gfpgan is None: print('>> GFPGAN not found. Face restoration is disabled.') - else: - image = self.gfpgan.process(image, strength, seed) + else: + image = self.gfpgan.process(image, strength, seed) + if facetool == 'codeformer': + if self.codeformer is None: + print('>> CodeFormer not found. Face restoration is disabled.') + else: + image = self.codeformer.process(image=image, strength=strength, device=self.device, seed=seed, fidelity=codeformer_fidelity) else: print(">> Face Restoration is disabled.") except Exception as e: