github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-07 13:32:09 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

SAEHD: fixed the use of pretrained liae model, now it produces less face morphing

Browse source
This commit is contained in:
Colombo 2020-03-27 00:15:18 +04:00
parent ca9138d6b7
commit 6687213fa5
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
models/Model_SAEHD/Model.py
View file

@ -483,7 +483,7 @@ class SAEHDModel(ModelBase):
if model == self.inter:
do_init = True
elif 'liae' in archi:
if model == self.inter_AB:
if model == self.inter_AB or model == self.inter_B:
do_init = True
else:
do_init = self.is_first_run()