Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

facexlib and codeformer are broken on mps currently, force CPU device.

Browse Source
This commit is contained in:
David Burnett 2022-09-24 14:33:12 +01:00 committed by Any-Winter-4079
parent e2bd492764
commit 636c356aaf
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
ldm/generate.py
View File

@ -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: