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

Merge branch 'invoke-ai:development' into development

Browse Source
This commit is contained in:
Any-Winter-4079
2022-09-24 22:01:00 +02:00
committed by GitHub
parent 3d2175c9f8 636c356aaf
commit c78c39e676
1 changed files with 2 additions and 1 deletions
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: