Fix same and equal

facefusion · Mar 3, 2025 · 0004a85 · 0004a85
1 parent d16d493
commit 0004a85
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/face_swapper/src/dataset.py b/face_swapper/src/dataset.py
@@ -55,12 +55,12 @@ def prepare_different_batch(self, source_path : str) -> Batch:
 		target_tensor = self.transforms(target_tensor)
 		return source_tensor, target_tensor
 
-	def prepare_same_batch(self, source_path : str) -> Batch:
+	def prepare_equal_batch(self, source_path : str) -> Batch:
 		source_tensor = io.read_image(source_path)
 		source_tensor = self.transforms(source_tensor)
 		return source_tensor, source_tensor
 
-	def prepare_equal_batch(self, source_path : str) -> Batch:
+	def prepare_same_batch(self, source_path : str) -> Batch:
 		target_directory_path = os.path.dirname(source_path)
 		target_file_name_and_extension = random.choice(os.listdir(target_directory_path))
 		target_path = os.path.join(target_directory_path, target_file_name_and_extension)