Add basic test for aad and unet

This commit is contained in:
henryruhs
2025-03-05 13:10:14 +01:00
parent 6fa8d6b6eb
commit d5c51a90e8
+2 -2
View File
@@ -1,8 +1,8 @@
import torch
import pytest
from ..src.networks.aad import AAD
from ..src.networks.unet import UNet
from face_swapper.src.networks.aad import AAD
from face_swapper.src.networks.unet import UNet
@pytest.mark.parametrize('output_size', [ 256 ])