From 9ede8a2a7def5b021f8f71f6362587418110a823 Mon Sep 17 00:00:00 2001 From: harisreedhar Date: Sun, 23 Mar 2025 18:38:48 +0530 Subject: [PATCH] changes --- face_swapper/src/models/generator.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/face_swapper/src/models/generator.py b/face_swapper/src/models/generator.py index 81c81d9..ab76154 100644 --- a/face_swapper/src/models/generator.py +++ b/face_swapper/src/models/generator.py @@ -19,12 +19,12 @@ class Generator(nn.Module): self.generator.apply(init_weight) self.masker.apply(init_weight) - def forward(self, source_embedding : Embedding, target_tensor : Tensor, target_features : Tuple[Feature, ...]) -> Tuple[Tensor, Mask, ]: + def forward(self, source_embedding : Embedding, target_tensor : Tensor, target_features : Tuple[Feature, ...]) -> Tuple[Tensor, Mask]: output_tensor = self.generator(source_embedding, target_features) target_feature = target_features[-1] output_mask = self.masker(target_tensor, target_feature) output_tensor = output_tensor * output_mask + target_tensor * (1 - output_mask) - return output_tensor, output_mask, + return output_tensor, output_mask def encode_features(self, input_tensor : Tensor) -> Tuple[Feature, ...]: return self.encoder(input_tensor)