fi0r1/roop
1
0
Fork 0
mirror of https://github.com/s0md3v/roop.git synced 2025-12-07 02:18:28 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
Moeblack 2023-06-12 21:06:34 +08:00 committed by henryruhs
parent 5e77a9eb17
commit 12ac4f7f61
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
roop/enhancer.py
View File

@ -96,7 +96,7 @@ def normalize_face(face):
return face_in_tensor.unsqueeze(0).to(device) return face_in_tensor.unsqueeze(0).to(device)
def enhance_face_in_tensor(face_in_tensor, codeformer_fidelity = 0.7): def enhance_face_in_tensor(face_in_tensor, codeformer_fidelity = 0.6):
with torch.no_grad(): with torch.no_grad():
enhanced_face_in_tensor = get_code_former()(face_in_tensor, w=codeformer_fidelity, adain=True)[0] enhanced_face_in_tensor = get_code_former()(face_in_tensor, w=codeformer_fidelity, adain=True)[0]
return enhanced_face_in_tensor return enhanced_face_in_tensor