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

Fix CoreML partially

Browse Source
This commit is contained in:
henryruhs 2024-11-25 20:26:57 +01:00
parent d73d75378f
commit 2adb716d15
2 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
facefusion/processors/modules/age_modifier.py
View File

@ -162,6 +162,9 @@ def forward(crop_vision_frame : VisionFrame, extend_vision_frame : VisionFrame,
age_modifier = get_inference_pool().get('age_modifier')
age_modifier_inputs = {}
if has_execution_provider('coreml'):
age_modifier.set_providers([ execution_provider_set.get('cpu') ])
for age_modifier_input in age_modifier.get_inputs():
if age_modifier_input.name == 'target':
age_modifier_inputs[age_modifier_input.name] = crop_vision_frame

5
facefusion/processors/modules/frame_colorizer.py
View File

@ -9,10 +9,8 @@ import facefusion.jobs.job_manager
import facefusion.jobs.job_store
import facefusion.processors.core as processors
from facefusion import config, content_analyser, inference_manager, logger, process_manager, state_manager, wording
from facefusion.choices import execution_provider_set
from facefusion.common_helper import create_int_metavar
from facefusion.download import conditional_download_hashes, conditional_download_sources, resolve_download_url
from facefusion.execution import has_execution_provider
from facefusion.filesystem import in_directory, is_image, is_video, resolve_relative_path, same_file_extension
from facefusion.processors import choices as processors_choices
from facefusion.processors.typing import FrameColorizerInputs
@ -199,9 +197,6 @@ def colorize_frame(temp_vision_frame : VisionFrame) -> VisionFrame:
def forward(color_vision_frame : VisionFrame) -> VisionFrame:
frame_colorizer = get_inference_pool().get('frame_colorizer')
if has_execution_provider('coreml'):
frame_colorizer.set_providers([ execution_provider_set.get('cpu') ])
with thread_semaphore():
color_vision_frame = frame_colorizer.run(None,
{