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

Prepare test for 512

Browse Source
This commit is contained in:
henryruhs
2025-03-05 13:50:23 +01:00
parent 64ebfa7b84
commit 2148e9b701
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
face_swapper/tests/test_networks.py
+8 -1
View File
@@ -7,7 +7,14 @@ from face_swapper.src.networks.unet import UNet
@pytest.mark.parametrize('output_size', [ 256 ])
def test_aad_with_unet(output_size : int) -> None:
generator = AAD(512, 4096, output_size, 2).eval()
identity_channels = 512
if output_size == 256:
output_channels = 4096
if output_size == 512:
output_channels = 8192
num_blocks = 2
generator = AAD(identity_channels, output_channels, output_size, num_blocks).eval()
encoder = UNet(output_size).eval()
source_tensor = torch.randn(1, 512)