From 65d1efa4040175a1ec8c424aba1ffb727097010b Mon Sep 17 00:00:00 2001
From: mike9251 <soapbox92@gmail.com>
Date: Fri, 15 Jul 2022 11:49:22 +0300
Subject: [PATCH] Fix ignored face_alingment parameter

---
 src/simswap.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/simswap.py b/src/simswap.py
index 154fda0..a4d1d81 100644
--- a/src/simswap.py
+++ b/src/simswap.py
@@ -134,7 +134,7 @@ def run_detect_align(self, image: np.ndarray, for_id: bool = False) -> Tuple[
             kps = kps[None, ...]
 
         align_imgs, transforms = align_face(image, kps, crop_size=self.crop_size,
-                                            mode=self.face_alignment_type)
+                                            mode='ffhq' if self.face_alignment_type == FaceAlignmentType.FFHQ else 'none')
 
         return align_imgs, transforms, detection.score