From cc9a0ba83e3b3f434953385d349f75c76b1f3363 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Mon, 3 Mar 2025 09:32:44 +0100 Subject: [PATCH] Add config to dataloader alignment --- face_swapper/src/dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/face_swapper/src/dataset.py b/face_swapper/src/dataset.py index 7bcde99..f1ad8b6 100644 --- a/face_swapper/src/dataset.py +++ b/face_swapper/src/dataset.py @@ -40,7 +40,7 @@ class DynamicDataset(Dataset[Tensor]): transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)) ]) - def warp_tensor(self, temp_tensor : Tensor): + def warp_tensor(self, temp_tensor : Tensor) -> Tensor: return warp_tensor(temp_tensor.unsqueeze(0), self.warp_matrix).squeeze(0) def prepare_different_batch(self, source_path : str) -> Batch: