From 5118c7229f80ecd5a0b89d91ae15387f8f95c3ff Mon Sep 17 00:00:00 2001
From: henryruhs <info@henryruhs.com>
Date: Thu, 24 Apr 2025 13:04:11 +0200
Subject: [PATCH] Rename arcface_128_v2 to arcface_128

---
 facefusion/face_helper.py                            | 2 +-
 facefusion/processors/modules/expression_restorer.py | 2 +-
 facefusion/processors/modules/face_debugger.py       | 2 +-
 facefusion/processors/modules/face_enhancer.py       | 2 +-
 facefusion/processors/modules/face_swapper.py        | 4 ++--
 facefusion/types.py                                  | 2 +-
 6 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/facefusion/face_helper.py b/facefusion/face_helper.py
index 6020f97..7db967a 100644
--- a/facefusion/face_helper.py
+++ b/facefusion/face_helper.py
@@ -25,7 +25,7 @@ WARP_TEMPLATE_SET : WarpTemplateSet =\
 		[ 0.37097589, 0.82469196 ],
 		[ 0.63151696, 0.82325089 ]
 	]),
-	'arcface_128_v2': numpy.array(
+	'arcface_128': numpy.array(
 	[
 		[ 0.36167656, 0.40387734 ],
 		[ 0.63696719, 0.40235469 ],
diff --git a/facefusion/processors/modules/expression_restorer.py b/facefusion/processors/modules/expression_restorer.py
index 0a49970..4d301a4 100755
--- a/facefusion/processors/modules/expression_restorer.py
+++ b/facefusion/processors/modules/expression_restorer.py
@@ -68,7 +68,7 @@ def create_static_model_set(download_scope : DownloadScope) -> ModelSet:
 					'path': resolve_relative_path('../.assets/models/live_portrait_generator.onnx')
 				}
 			},
-			'template': 'arcface_128_v2',
+			'template': 'arcface_128',
 			'size': (512, 512)
 		}
 	}
diff --git a/facefusion/processors/modules/face_debugger.py b/facefusion/processors/modules/face_debugger.py
index 9a544c2..e4397f2 100755
--- a/facefusion/processors/modules/face_debugger.py
+++ b/facefusion/processors/modules/face_debugger.py
@@ -90,7 +90,7 @@ def debug_face(target_face : Face, temp_vision_frame : VisionFrame) -> VisionFra
 			cv2.line(temp_vision_frame, (x1, y1), (x1, y2), primary_light_color, 3)
 
 	if 'face-mask' in face_debugger_items:
-		crop_vision_frame, affine_matrix = warp_face_by_face_landmark_5(temp_vision_frame, target_face.landmark_set.get('5/68'), 'arcface_128_v2', (512, 512))
+		crop_vision_frame, affine_matrix = warp_face_by_face_landmark_5(temp_vision_frame, target_face.landmark_set.get('5/68'), 'arcface_128', (512, 512))
 		inverse_matrix = cv2.invertAffineTransform(affine_matrix)
 		temp_size = temp_vision_frame.shape[:2][::-1]
 		crop_masks = []
diff --git a/facefusion/processors/modules/face_enhancer.py b/facefusion/processors/modules/face_enhancer.py
index 4e71a58..1787d69 100755
--- a/facefusion/processors/modules/face_enhancer.py
+++ b/facefusion/processors/modules/face_enhancer.py
@@ -131,7 +131,7 @@ def create_static_model_set(download_scope : DownloadScope) -> ModelSet:
 					'path': resolve_relative_path('../.assets/models/gpen_bfr_256.onnx')
 				}
 			},
-			'template': 'arcface_128_v2',
+			'template': 'arcface_128',
 			'size': (256, 256)
 		},
 		'gpen_bfr_512':
diff --git a/facefusion/processors/modules/face_swapper.py b/facefusion/processors/modules/face_swapper.py
index 791a308..7ce0ed4 100755
--- a/facefusion/processors/modules/face_swapper.py
+++ b/facefusion/processors/modules/face_swapper.py
@@ -211,7 +211,7 @@ def create_static_model_set(download_scope : DownloadScope) -> ModelSet:
 				}
 			},
 			'type': 'inswapper',
-			'template': 'arcface_128_v2',
+			'template': 'arcface_128',
 			'size': (128, 128),
 			'mean': [ 0.0, 0.0, 0.0 ],
 			'standard_deviation': [ 1.0, 1.0, 1.0 ]
@@ -235,7 +235,7 @@ def create_static_model_set(download_scope : DownloadScope) -> ModelSet:
 				}
 			},
 			'type': 'inswapper',
-			'template': 'arcface_128_v2',
+			'template': 'arcface_128',
 			'size': (128, 128),
 			'mean': [ 0.0, 0.0, 0.0 ],
 			'standard_deviation': [ 1.0, 1.0, 1.0 ]
diff --git a/facefusion/types.py b/facefusion/types.py
index 7f7cc0a..def3aa0 100755
--- a/facefusion/types.py
+++ b/facefusion/types.py
@@ -87,7 +87,7 @@ Content : TypeAlias = Dict[str, Any]
 
 Commands : TypeAlias = List[str]
 
-WarpTemplate = Literal['arcface_112_v1', 'arcface_112_v2', 'arcface_128_v2', 'dfl_whole_face', 'ffhq_512', 'mtcnn_512', 'styleganex_384']
+WarpTemplate = Literal['arcface_112_v1', 'arcface_112_v2', 'arcface_128', 'dfl_whole_face', 'ffhq_512', 'mtcnn_512', 'styleganex_384']
 WarpTemplateSet : TypeAlias = Dict[WarpTemplate, NDArray[Any]]
 ProcessMode = Literal['output', 'preview', 'stream']