CalvinBackup/facefusion-labs
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion-labs.git synced 2026-08-31 00:10:39 +02:00
Code Issues Packages Projects Releases Wiki Activity

Add basic test for aad and unet

Browse Source
This commit is contained in:
henryruhs
2025-03-11 14:43:10 +01:00
parent 72371b9f11
commit 6fa8d6b6eb
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/ci.yml
+1 -1
View File
@@ -31,4 +31,4 @@ jobs:
python-version: '3.12' python-version: '3.12'
- run: pip install torch torchvision - run: pip install torch torchvision
- run: pip install pytest - run: pip install pytest
- run: pytest - run: pytest face_swapper/tests
face_swapper/tests/test_networks.py
+2 -2
View File
@@ -1,8 +1,8 @@
import torch import torch
import pytest import pytest
from face_swapper.src.networks.aad import AAD from ..src.networks.aad import AAD
from face_swapper.src.networks.unet import UNet from ..src.networks.unet import UNet
@pytest.mark.parametrize('output_size', [ 256 ]) @pytest.mark.parametrize('output_size', [ 256 ])