From c8801ececd1ad5b30c8b16efea2a8b5d94572724 Mon Sep 17 00:00:00 2001
From: henryruhs <info@henryruhs.com>
Date: Wed, 26 Feb 2025 16:56:02 +0100
Subject: [PATCH] Fix export using Trainer

---
 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 29fa196..e016b80 100644
--- a/face_swapper/src/training.py
+++ b/face_swapper/src/training.py
@@ -40,7 +40,7 @@ class FaceSwapperTrainer(lightning.LightningModule):
 		self.embedder = torch.jit.load(embedder_path, map_location = 'cpu') # type:ignore[no-untyped-call]
 		self.automatic_optimization = False
 
-	def forward(self, target_tensor : Tensor, source_embedding : Embedding) -> Tensor:
+	def forward(self, source_embedding : Embedding, target_tensor : Tensor) -> Tensor:
 		output_tensor = self.generator(source_embedding, target_tensor)
 		return output_tensor