From f989df39e9b2183e4693f1aff692fca628698d45 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Tue, 11 Mar 2025 23:03:22 +0100 Subject: [PATCH] Fix test --- face_swapper/src/networks/aad.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/face_swapper/src/networks/aad.py b/face_swapper/src/networks/aad.py index 9e24b03..7babfbe 100644 --- a/face_swapper/src/networks/aad.py +++ b/face_swapper/src/networks/aad.py @@ -65,7 +65,7 @@ class AAD(nn.Module): temp_tensor = layer(temp_tensors, source_embedding, target_attribute) temp_tensors = nn.functional.interpolate(temp_tensor, scale_factor = 2, mode = 'bilinear', align_corners = False) - target_attribute = target_attributes[-1], + target_attribute = target_attributes[-1] temp_tensors = self.layers[-1](temp_tensors, source_embedding, target_attribute) output_tensor = torch.tanh(temp_tensors) return output_tensor