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

Make CI happy

Browse Source
This commit is contained in:
henryruhs 2023-10-14 23:41:48 +02:00
parent 27e506f5ac
commit dc78e3b2ac
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
facefusion/predictor.py
View File

@ -17,7 +17,7 @@ PREDICTOR = None
THREAD_LOCK : threading.Lock = threading.Lock()
NAME = 'FACEFUSION.PREDICTOR'
MODEL_URL = 'https://github.com/facefusion/facefusion-assets/releases/download/models/open_nsfw.onnx'
MODEL_PATH = resolve_relative_path('../.assets/models/_open_nsfw.onnx')
MODEL_PATH = resolve_relative_path('../.assets/models/open_nsfw.onnx')
MAX_PROBABILITY = 0.80
STREAM_COUNTER = 0

3
facefusion/processors/frame/modules/face_enhancer.py
View File

@ -6,6 +6,7 @@ import numpy
import onnxruntime
import facefusion.globals
import facefusion.processors.frame.core as frame_processors
from facefusion import wording
from facefusion.face_analyser import get_many_faces, clear_face_analyser
from facefusion.face_helper import warp_face, paste_back
@ -191,4 +192,4 @@ def process_image(source_path : str, target_path : str, output_path : str) -> No
def process_video(source_path : str, temp_frame_paths : List[str]) -> None:
facefusion.processors.frame.core.multi_process_frames(None, temp_frame_paths, process_frames)
frame_processors.multi_process_frames(None, temp_frame_paths, process_frames)