From 636c356aafdeceab646648bcb25ba38c16a49e32 Mon Sep 17 00:00:00 2001 From: David Burnett Date: Sat, 24 Sep 2022 14:33:12 +0100 Subject: [PATCH] facexlib and codeformer are broken on mps currently, force CPU device. --- ldm/generate.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/ldm/generate.py b/ldm/generate.py index e477e1d9ae..6dfbfdc018 100644 --- a/ldm/generate.py +++ b/ldm/generate.py @@ -742,7 +742,8 @@ class Generate: 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) + cf_device = 'cpu' if str(self.device) == 'mps' else self.device + image = self.codeformer.process(image=image, strength=strength, device=cf_device, seed=seed, fidelity=codeformer_fidelity) else: print(">> Face Restoration is disabled.") except Exception as e: