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

Use /tmp/facefusion

Browse Source
This commit is contained in:
henryruhs
2023-08-20 19:48:52 +02:00
parent 26d3b634f3
commit 48a068512d
2 changed files with 5 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
facefusion/core.py
View File

@@ -190,7 +190,7 @@ def process_video() -> None:
else: else:
update_status(wording.get('restoring_audio_issues')) update_status(wording.get('restoring_audio_issues'))
restore_audio(facefusion.globals.target_path, facefusion.globals.output_path) restore_audio(facefusion.globals.target_path, facefusion.globals.output_path)
# clean temp # clear temp
update_status(wording.get('clearing_temp')) update_status(wording.get('clearing_temp'))
clear_temp(facefusion.globals.target_path) clear_temp(facefusion.globals.target_path)
# validate video # validate video

9
facefusion/utilities.py
View File

@@ -16,8 +16,8 @@ from tqdm import tqdm
import facefusion.globals import facefusion.globals
from facefusion import wording from facefusion import wording
TEMP_DIRECTORY = tempfile.mkdtemp(prefix = 'facefusion') TEMP_DIRECTORY_PATH = os.path.join(tempfile.gettempdir(), 'facefusion')
TEMP_VIDEO_FILE = 'temp.mp4' TEMP_OUTPUT_NAME = 'temp.mp4'
# monkey patch ssl # monkey patch ssl
if platform.system().lower() == 'darwin': if platform.system().lower() == 'darwin':
@@ -99,13 +99,12 @@ def get_temp_frame_paths(target_path : str) -> List[str]:
def get_temp_directory_path(target_path : str) -> str: def get_temp_directory_path(target_path : str) -> str:
target_name, _ = os.path.splitext(os.path.basename(target_path)) target_name, _ = os.path.splitext(os.path.basename(target_path))
target_directory_path = os.path.dirname(target_path) return os.path.join(TEMP_DIRECTORY_PATH, target_name)
return os.path.join(target_directory_path, TEMP_DIRECTORY, target_name)
def get_temp_output_path(target_path : str) -> str: def get_temp_output_path(target_path : str) -> str:
temp_directory_path = get_temp_directory_path(target_path) temp_directory_path = get_temp_directory_path(target_path)
return os.path.join(temp_directory_path, TEMP_VIDEO_FILE) return os.path.join(temp_directory_path, TEMP_OUTPUT_NAME)
def normalize_output_path(source_path : str, target_path : str, output_path : str) -> Optional[str]: def normalize_output_path(source_path : str, target_path : str, output_path : str) -> Optional[str]: