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

Improve model file handling part2

Browse Source
This commit is contained in:
henryruhs 2024-04-01 23:55:20 +02:00
parent 73d8af4258
commit 2344ca2c3a
5 changed files with 20 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -160,12 +160,12 @@ def post_check() -> bool:
model_url = get_options('model').get('url')
model_path = get_options('model').get('path')
if facefusion.globals.skip_download and not is_file(model_path):
logger.error(wording.get('model_file_not_present') + wording.get('exclamation_mark'), NAME)
return False
if not is_download_done(model_url, model_path):
if not facefusion.globals.skip_download and not is_download_done(model_url, model_path):
logger.error(wording.get('model_download_not_done') + wording.get('exclamation_mark'), NAME)
return False
if not is_file(model_path):
logger.error(wording.get('model_file_not_present') + wording.get('exclamation_mark'), NAME)
return False
return True

8
facefusion/processors/frame/modules/face_swapper.py
View File

@ -188,12 +188,12 @@ def post_check() -> bool:
model_url = get_options('model').get('url')
model_path = get_options('model').get('path')
if facefusion.globals.skip_download and not is_file(model_path):
logger.error(wording.get('model_file_not_present') + wording.get('exclamation_mark'), NAME)
return False
if not is_download_done(model_url, model_path):
if not facefusion.globals.skip_download and not is_download_done(model_url, model_path):
logger.error(wording.get('model_download_not_done') + wording.get('exclamation_mark'), NAME)
return False
if not is_file(model_path):
logger.error(wording.get('model_file_not_present') + wording.get('exclamation_mark'), NAME)
return False
return True

8
facefusion/processors/frame/modules/frame_colorizer.py
View File

@ -114,12 +114,12 @@ def post_check() -> bool:
model_url = get_options('model').get('url')
model_path = get_options('model').get('path')
if facefusion.globals.skip_download and not is_file(model_path):
logger.error(wording.get('model_file_not_present') + wording.get('exclamation_mark'), NAME)
return False
if not is_download_done(model_url, model_path):
if not facefusion.globals.skip_download and not is_download_done(model_url, model_path):
logger.error(wording.get('model_download_not_done') + wording.get('exclamation_mark'), NAME)
return False
if not is_file(model_path):
logger.error(wording.get('model_file_not_present') + wording.get('exclamation_mark'), NAME)
return False
return True

8
facefusion/processors/frame/modules/frame_enhancer.py
View File

@ -135,12 +135,12 @@ def post_check() -> bool:
model_url = get_options('model').get('url')
model_path = get_options('model').get('path')
if facefusion.globals.skip_download and not is_file(model_path):
logger.error(wording.get('model_file_not_present') + wording.get('exclamation_mark'), NAME)
return False
if not is_download_done(model_url, model_path):
if not facefusion.globals.skip_download and not is_download_done(model_url, model_path):
logger.error(wording.get('model_download_not_done') + wording.get('exclamation_mark'), NAME)
return False
if not is_file(model_path):
logger.error(wording.get('model_file_not_present') + wording.get('exclamation_mark'), NAME)
return False
return True

8
facefusion/processors/frame/modules/lip_syncer.py
View File

@ -101,12 +101,12 @@ def post_check() -> bool:
model_url = get_options('model').get('url')
model_path = get_options('model').get('path')
if facefusion.globals.skip_download and not is_file(model_path):
logger.error(wording.get('model_file_not_present') + wording.get('exclamation_mark'), NAME)
return False
if not is_download_done(model_url, model_path):
if not facefusion.globals.skip_download and not is_download_done(model_url, model_path):
logger.error(wording.get('model_download_not_done') + wording.get('exclamation_mark'), NAME)
return False
if not is_file(model_path):
logger.error(wording.get('model_file_not_present') + wording.get('exclamation_mark'), NAME)
return False
return True