From 904a447e06f917fab291fe8007a70a80680af13e Mon Sep 17 00:00:00 2001 From: henryruhs Date: Fri, 14 Mar 2025 08:30:55 +0100 Subject: [PATCH] Mask typing and naming related updates --- face_swapper/src/training.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/face_swapper/src/training.py b/face_swapper/src/training.py index 16f1790..502c907 100644 --- a/face_swapper/src/training.py +++ b/face_swapper/src/training.py @@ -53,7 +53,7 @@ class FaceSwapperTrainer(LightningModule): self.mask_loss = MaskLoss(config_parser, self.face_parser) self.automatic_optimization = False - def forward(self, source_embedding : Embedding, target_tensor : Tensor) -> Tuple[Tensor, Tensor]: + def forward(self, source_embedding : Embedding, target_tensor : Tensor) -> Tuple[Tensor, Mask]: with torch.no_grad(): output_tensor, target_attributes = self.generator(source_embedding, target_tensor) target_attribute = target_attributes[-1]