From a7a21cd684e7d9859c99339ffe77984b93004623 Mon Sep 17 00:00:00 2001 From: harisreedhar Date: Tue, 11 Mar 2025 17:42:29 +0530 Subject: [PATCH] changes --- face_swapper/src/helper.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/face_swapper/src/helper.py b/face_swapper/src/helper.py index 7fc5e9e..015ba43 100644 --- a/face_swapper/src/helper.py +++ b/face_swapper/src/helper.py @@ -39,7 +39,7 @@ def calc_embedding(embedder : EmbedderModule, input_tensor : Tensor, padding : P def overlay_mask(target_tensor : Tensor, mask_tensor : Tensor) -> Tensor: - color_tensor = torch.zeros(*list(target_tensor.shape), dtype = target_tensor.dtype, device = target_tensor.device) + color_tensor = torch.zeros(*target_tensor.shape, dtype = target_tensor.dtype, device = target_tensor.device) color_tensor[:, 2, :, :] = 1 mask_tensor = mask_tensor.repeat(1, 3, 1, 1).clamp(0, 0.8) output_tensor = target_tensor * (1 - mask_tensor) + color_tensor * mask_tensor