mirrors/facefusion
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Use static model set creation

Browse Source
This commit is contained in:
henryruhs 2024-11-19 22:46:38 +01:00
parent 01420056b4
commit ba874607d7
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
facefusion/processors/modules/face_swapper.py
View File

@ -1,4 +1,5 @@
from argparse import ArgumentParser from argparse import ArgumentParser
from functools import lru_cache
from typing import List, Tuple from typing import List, Tuple
import numpy import numpy
@ -26,7 +27,8 @@ from facefusion.typing import ApplyStateItem, Args, Embedding, Face, InferencePo
from facefusion.vision import read_image, read_static_image, read_static_images, unpack_resolution, write_image from facefusion.vision import read_image, read_static_image, read_static_images, unpack_resolution, write_image
def create_model_set() -> ModelSet: @lru_cache(maxsize = None)
def create_static_model_set() -> ModelSet:
return\ return\
{ {
'blendswap_256': 'blendswap_256':
@ -346,7 +348,7 @@ def clear_inference_pool() -> None:
def get_model_options() -> ModelOptions: def get_model_options() -> ModelOptions:
face_swapper_model = state_manager.get_item('face_swapper_model') face_swapper_model = state_manager.get_item('face_swapper_model')
face_swapper_model = 'inswapper_128' if has_execution_provider('coreml') and face_swapper_model == 'inswapper_128_fp16' else face_swapper_model face_swapper_model = 'inswapper_128' if has_execution_provider('coreml') and face_swapper_model == 'inswapper_128_fp16' else face_swapper_model
return create_model_set().get(face_swapper_model) return create_static_model_set().get(face_swapper_model)
def register_args(program : ArgumentParser) -> None: def register_args(program : ArgumentParser) -> None: