diff --git a/facefusion/processors/choices.py b/facefusion/processors/choices.py index 89c87138..c337d85a 100755 --- a/facefusion/processors/choices.py +++ b/facefusion/processors/choices.py @@ -4,7 +4,7 @@ from facefusion.common_helper import create_float_range, create_int_range from facefusion.processors.typing import AgeModifierModel, DeepSwapperModel, ExpressionRestorerModel, FaceDebuggerItem, FaceEditorModel, FaceEnhancerModel, FaceSwapperSet, FrameColorizerModel, FrameEnhancerModel, LipSyncerModel age_modifier_models : List[AgeModifierModel] = [ 'styleganex_age' ] -deep_swapper_models : List[DeepSwapperModel] = [ 'jackie_chan' ] +deep_swapper_models : List[DeepSwapperModel] = [ 'iperov/jackie_chan_224' ] expression_restorer_models : List[ExpressionRestorerModel] = [ 'live_portrait' ] face_debugger_items : List[FaceDebuggerItem] = [ 'bounding-box', 'face-landmark-5', 'face-landmark-5/68', 'face-landmark-68', 'face-landmark-68/5', 'face-mask', 'face-detector-score', 'face-landmarker-score', 'age', 'gender', 'race' ] face_editor_models : List[FaceEditorModel] = [ 'live_portrait' ] diff --git a/facefusion/processors/modules/deep_swapper.py b/facefusion/processors/modules/deep_swapper.py index 117c79be..5700ab13 100755 --- a/facefusion/processors/modules/deep_swapper.py +++ b/facefusion/processors/modules/deep_swapper.py @@ -24,7 +24,7 @@ from facefusion.vision import conditional_match_frame_color, read_image, read_st MODEL_SET : ModelSet =\ { - 'jackie_chan': + 'iperov/jackie_chan_224': { 'hashes': { @@ -67,7 +67,7 @@ def get_model_options() -> ModelOptions: def register_args(program : ArgumentParser) -> None: group_processors = find_argument_group(program, 'processors') if group_processors: - group_processors.add_argument('--deep-swapper-model', help = wording.get('help.deep_swapper_model'), default = config.get_str_value('processors.deep_swapper_model', 'jackie_chan'), choices = processors_choices.deep_swapper_models) + group_processors.add_argument('--deep-swapper-model', help = wording.get('help.deep_swapper_model'), default = config.get_str_value('processors.deep_swapper_model', 'iperov/jackie_chan_224'), choices = processors_choices.deep_swapper_models) facefusion.jobs.job_store.register_step_keys([ 'deep_swapper_model' ]) diff --git a/facefusion/processors/typing.py b/facefusion/processors/typing.py index b52138b4..3f86ea8e 100644 --- a/facefusion/processors/typing.py +++ b/facefusion/processors/typing.py @@ -5,7 +5,7 @@ from numpy._typing import NDArray from facefusion.typing import AppContext, AudioFrame, Face, FaceSet, VisionFrame AgeModifierModel = Literal['styleganex_age'] -DeepSwapperModel = Literal['jackie_chan'] +DeepSwapperModel = Literal['iperov/jackie_chan_224'] ExpressionRestorerModel = Literal['live_portrait'] FaceDebuggerItem = Literal['bounding-box', 'face-landmark-5', 'face-landmark-5/68', 'face-landmark-68', 'face-landmark-68/5', 'face-mask', 'face-detector-score', 'face-landmarker-score', 'age', 'gender', 'race'] FaceEditorModel = Literal['live_portrait']