mirror of
https://github.com/facefusion/facefusion-labs.git
synced 2026-04-19 15:56:37 +02:00
Add basic test for aad and unet
This commit is contained in:
@@ -31,4 +31,4 @@ jobs:
|
||||
python-version: '3.12'
|
||||
- run: pip install torch torchvision
|
||||
- run: pip install pytest
|
||||
- run: pytest
|
||||
- run: pytest face_swapper/tests
|
||||
|
||||
@@ -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 ])
|
||||
|
||||
Reference in New Issue
Block a user