CalvinBackup/facefusion-labs
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion-labs.git synced 2026-05-22 23:59:40 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix CI

Browse Source
This commit is contained in:
henryruhs
2025-03-05 15:02:12 +01:00
parent 866019d44f
commit f791178ded
2 changed files with 8 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/ci.yml
+1 -1
View File
@@ -28,4 +28,4 @@ jobs:
python-version: '3.12'
- run: pip install torch torchvision
- run: pip install pytest
- run: PYTHONPATH=/Users/runner/work/facefusion-labs/facefusion-labs pytest
- run: PYTHONPATH=/home/runner/work/facefusion-labs/facefusion-labs pytest
face_swapper/tests/test_networks.py
+7 -14
View File
@@ -1,14 +1,8 @@
import os
import sys
import pytest
import torch
package_directory = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
sys.path.append(package_directory)
#from face_swapper.src.networks.aad import AAD
#from face_swapper.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 ])
@@ -20,14 +14,13 @@ def test_aad_with_unet(output_size : int) -> None:
output_channels = 8192
num_blocks = 2
#generator = AAD(identity_channels, output_channels, output_size, num_blocks).eval()
#encoder = UNet(output_size).eval()
generator = AAD(identity_channels, output_channels, output_size, num_blocks).eval()
encoder = UNet(output_size).eval()
source_tensor = torch.randn(1, 512)
target_tensor = torch.randn(1, 3, output_size, output_size)
#target_attributes = encoder(target_tensor)
#output_tensor = generator(source_tensor, target_attributes)
target_attributes = encoder(target_tensor)
output_tensor = generator(source_tensor, target_attributes)
assert package_directory == None
#assert output_tensor.shape == (1, 3, output_size, output_size)
assert output_tensor.shape == (1, 3, output_size, output_size)