Release five frame enhancer models

facefusion/processors/choices.py

@@ -24,7 +24,7 @@ face_swapper_set : FaceSwapperSet =\
 }
 frame_colorizer_models : List[FrameColorizerModel] = [ 'ddcolor', 'ddcolor_artistic', 'deoldify', 'deoldify_artistic', 'deoldify_stable' ]
 frame_colorizer_sizes : List[str] = [ '192x192', '256x256', '384x384', '512x512' ]
-frame_enhancer_models : List[FrameEnhancerModel] = [ 'clear_reality_x4', 'lsdir_x4', 'nomos8k_sc_x4', 'real_esrgan_x2', 'real_esrgan_x2_fp16', 'real_esrgan_x4', 'real_esrgan_x4_fp16', 'real_esrgan_x8', 'real_esrgan_x8_fp16', 'real_hatgan_x4', 'span_kendata_x4', 'ultra_sharp_x4' ]
+frame_enhancer_models : List[FrameEnhancerModel] = [ 'clear_reality_x4', 'foolhardy_remacri_x4', 'lsdir_x4', 'nmkd_siax_x4', 'nomos8k_sc_x4', 'real_esrgan_x2', 'real_esrgan_x2_fp16', 'real_esrgan_x4', 'real_esrgan_x4_fp16', 'real_esrgan_x8', 'real_esrgan_x8_fp16', 'real_hatgan_x4', 'real_web_photo_x4', 'realistic_rescaler_x4', 'span_kendata_x4', 'swin2_sr_x4', 'ultra_sharp_x4' ]
 lip_syncer_models : List[LipSyncerModel] = [ 'wav2lip_96', 'wav2lip_gan_96' ]
 
 age_modifier_direction_range : Sequence[int] = create_int_range(-100, 100, 1)

facefusion/processors/modules/frame_enhancer.py

@@ -41,6 +41,27 @@ MODEL_SET : ModelSet =\
 		'size': (128, 8, 4),
 		'scale': 4
 	},
+	'foolhardy_remacri_x4':
+	{
+		'hashes':
+		{
+			'frame_enhancer':
+			{
+				'url': 'https://github.com/facefusion/facefusion-assets/releases/download/models-3.1.0/foolhardy_remacri_x4.hash',
+				'path': resolve_relative_path('../.assets/models/foolhardy_remacri_x4.hash')
+			}
+		},
+		'sources':
+		{
+			'frame_enhancer':
+			{
+				'url': 'https://github.com/facefusion/facefusion-assets/releases/download/models-3.1.0/foolhardy_remacri_x4.onnx',
+				'path': resolve_relative_path('../.assets/models/foolhardy_remacri_x4.onnx')
+			}
+		},
+		'size': (128, 8, 4),
+		'scale': 4
+	},
 	'lsdir_x4':
 	{
 		'hashes':
@@ -62,6 +83,27 @@ MODEL_SET : ModelSet =\
 		'size': (128, 8, 4),
 		'scale': 4
 	},
+	'nmkd_siax_x4':
+	{
+		'hashes':
+		{
+			'frame_enhancer':
+			{
+				'url': 'https://github.com/facefusion/facefusion-assets/releases/download/models-3.1.0/nmkd_siax_x4.hash',
+				'path': resolve_relative_path('../.assets/models/nmkd_siax_x4.hash')
+			}
+		},
+		'sources':
+		{
+			'frame_enhancer':
+			{
+				'url': 'https://github.com/facefusion/facefusion-assets/releases/download/models-3.1.0/nmkd_siax_x4.onnx',
+				'path': resolve_relative_path('../.assets/models/nmkd_siax_x4.onnx')
+			}
+		},
+		'size': (128, 8, 4),
+		'scale': 4
+	},
 	'nomos8k_sc_x4':
 	{
 		'hashes':
@@ -230,6 +272,48 @@ MODEL_SET : ModelSet =\
 		'size': (256, 16, 8),
 		'scale': 4
 	},
+	'real_web_photo_x4':
+	{
+		'hashes':
+		{
+			'frame_enhancer':
+			{
+				'url': 'https://github.com/facefusion/facefusion-assets/releases/download/models-3.1.0/real_web_photo_x4.hash',
+				'path': resolve_relative_path('../.assets/models/real_web_photo_x4.hash')
+			}
+		},
+		'sources':
+		{
+			'frame_enhancer':
+			{
+				'url': 'https://github.com/facefusion/facefusion-assets/releases/download/models-3.1.0/real_web_photo_x4.onnx',
+				'path': resolve_relative_path('../.assets/models/real_web_photo_x4.onnx')
+			}
+		},
+		'size': (64, 4, 2),
+		'scale': 4
+	},
+	'realistic_rescaler_x4':
+	{
+		'hashes':
+		{
+			'frame_enhancer':
+			{
+				'url': 'https://github.com/facefusion/facefusion-assets/releases/download/models-3.1.0/realistic_rescaler_x4.hash',
+				'path': resolve_relative_path('../.assets/models/realistic_rescaler_x4.hash')
+			}
+		},
+		'sources':
+		{
+			'frame_enhancer':
+			{
+				'url': 'https://github.com/facefusion/facefusion-assets/releases/download/models-3.1.0/realistic_rescaler_x4.onnx',
+				'path': resolve_relative_path('../.assets/models/realistic_rescaler_x4.onnx')
+			}
+		},
+		'size': (128, 8, 4),
+		'scale': 4
+	},
 	'span_kendata_x4':
 	{
 		'hashes':
@@ -251,6 +335,27 @@ MODEL_SET : ModelSet =\
 		'size': (128, 8, 4),
 		'scale': 4
 	},
+	'swin2_sr_x4':
+	{
+		'hashes':
+		{
+			'frame_enhancer':
+			{
+				'url': 'https://github.com/facefusion/facefusion-assets/releases/download/models-3.1.0/swin2_sr_x4.hash',
+				'path': resolve_relative_path('../.assets/models/swin2_sr_x4.hash')
+			}
+		},
+		'sources':
+		{
+			'frame_enhancer':
+			{
+				'url': 'https://github.com/facefusion/facefusion-assets/releases/download/models-3.1.0/swin2_sr_x4.onnx',
+				'path': resolve_relative_path('../.assets/models/swin2_sr_x4.onnx')
+			}
+		},
+		'size': (128, 8, 4),
+		'scale': 4
+	},
 	'ultra_sharp_x4':
 	{
 		'hashes':

facefusion/processors/typing.py

@@ -12,7 +12,7 @@ FaceEditorModel = Literal['live_portrait']
 FaceEnhancerModel = Literal['codeformer', 'gfpgan_1.2', 'gfpgan_1.3', 'gfpgan_1.4', 'gpen_bfr_256', 'gpen_bfr_512', 'gpen_bfr_1024', 'gpen_bfr_2048', 'restoreformer_plus_plus']
 FaceSwapperModel = Literal['blendswap_256', 'ghost_1_256', 'ghost_2_256', 'ghost_3_256', 'hififace_unofficial_256', 'inswapper_128', 'inswapper_128_fp16', 'simswap_256', 'simswap_unofficial_512', 'uniface_256']
 FrameColorizerModel = Literal['ddcolor', 'ddcolor_artistic', 'deoldify', 'deoldify_artistic', 'deoldify_stable']
-FrameEnhancerModel = Literal['clear_reality_x4', 'lsdir_x4', 'nomos8k_sc_x4', 'real_esrgan_x2', 'real_esrgan_x2_fp16', 'real_esrgan_x4', 'real_esrgan_x4_fp16', 'real_hatgan_x4', 'real_esrgan_x8', 'real_esrgan_x8_fp16', 'span_kendata_x4', 'ultra_sharp_x4']
+FrameEnhancerModel = Literal['clear_reality_x4', 'foolhardy_remacri_x4', 'lsdir_x4', 'nmkd_siax_x4', 'nomos8k_sc_x4', 'real_esrgan_x2', 'real_esrgan_x2_fp16', 'real_esrgan_x4', 'real_esrgan_x4_fp16', 'real_esrgan_x8', 'real_esrgan_x8_fp16', 'real_hatgan_x4', 'real_web_photo_x4', 'realistic_rescaler_x4', 'span_kendata_x4', 'swin2_sr_x4', 'ultra_sharp_x4']
 LipSyncerModel = Literal['wav2lip_96', 'wav2lip_gan_96']
 
 FaceSwapperSet = Dict[FaceSwapperModel, List[str]]