Add basic test for aad and unet

This commit is contained in:
henryruhs
2025-03-05 12:57:15 +01:00
parent 72371b9f11
commit 6fa8d6b6eb
2 changed files with 3 additions and 3 deletions
+2 -2
View File
@@ -1,8 +1,8 @@
import torch
import pytest
from face_swapper.src.networks.aad import AAD
from face_swapper.src.networks.unet import UNet
from ..src.networks.aad import AAD
from ..src.networks.unet import UNet
@pytest.mark.parametrize('output_size', [ 256 ])