CalvinBackup/facefusion-labs
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion-labs.git synced 2026-05-22 23:59:40 +02:00
Code Issues Packages Projects Releases Wiki Activity

Minor adjustment for test

Browse Source
This commit is contained in:
henryruhs
2025-03-06 22:28:54 +01:00
parent 7e938c2ec9
commit e5f983b2bf
1 changed files with 9 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
face_swapper/tests/test_networks.py
+9 -6
View File
@@ -10,13 +10,16 @@ from face_swapper.src.networks.unet import UNet
@pytest.mark.parametrize('output_size', [ 128, 256, 512 ])
def test_aad_with_unet(output_size : int) -> None:
config_parser = ConfigParser()
config_parser['training.model.generator'] =\
config_parser.read_dict(
{
'identity_channels': '512',
'output_channels': str(output_size * 16),
'output_size': str(output_size),
'num_blocks': '2'
}
'training.model.generator':
{
'identity_channels': '512',
'output_channels': str(output_size * 16),
'output_size': str(output_size),
'num_blocks': '2'
}
})
generator = AAD(config_parser).eval()
encoder = UNet(config_parser).eval()