Crazy fix

This commit is contained in:
henryruhs
2025-03-05 14:42:21 +01:00
parent f678aa8f7e
commit 5f053f9f69
+1
View File
@@ -4,6 +4,7 @@ import torch
from face_swapper.src.networks.aad import AAD
from face_swapper.src.networks.unet import UNet
@pytest.mark.parametrize('output_size', [ 256 ])
def test_aad_with_unet(output_size : int) -> None:
identity_channels = 512