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

fix: check gan version for certain gan settings

Browse source
This commit is contained in:
jh 2021-03-21 23:46:13 -07:00
commit ba1a1b2b91
1 changed files with 6 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

1
models/Model_SAEHD/Model.py
View file

@ -158,6 +158,7 @@ Examples: df, liae, df-d, df-ud, liae-ud, ...
gan_version = np.clip (io.input_int("GAN version", default_gan_version, add_info="2 or 3", help_message="Choose GAN version (v2: 7/16/2020, v3: 1/3/2021):"), 2, 3)
self.options['gan_version'] = gan_version
if self.options['gan_version'] == 3:
gan_patch_size = np.clip ( io.input_int("GAN patch size", default_gan_patch_size, add_info="3-640", help_message="The higher patch size, the higher the quality, the more VRAM is required. You can get sharper edges even at the lowest setting. Typical fine value is resolution / 8." ), 3, 640 )
self.options['gan_patch_size'] = gan_patch_size