Fix face_enhancer issues

This commit is contained in:
Joep de Jong 2024-07-07 14:33:32 +02:00
parent c21ebb1b67
commit db97940adb
No known key found for this signature in database
GPG Key ID: 76A2026723C827A4
2 changed files with 5 additions and 4 deletions

View File

@ -25,6 +25,7 @@ def load_frame_processor_module(frame_processor: str) -> Any:
if not hasattr(frame_processor_module, method_name):
sys.exit()
except ImportError:
print(f"Frame processor {frame_processor} not found")
sys.exit()
return frame_processor_module
@ -47,8 +48,8 @@ def set_frame_processors_modules_from_ui(frame_processors: List[str]) -> None:
FRAME_PROCESSORS_MODULES.append(frame_processor_module)
modules.globals.frame_processors.append(frame_processor)
if state == False:
frame_processor_module = load_frame_processor_module(frame_processor)
try:
frame_processor_module = load_frame_processor_module(frame_processor)
FRAME_PROCESSORS_MODULES.remove(frame_processor_module)
modules.globals.frame_processors.remove(frame_processor)
except:

View File

@ -1,6 +1,6 @@
--extra-index-url https://download.pytorch.org/whl/cu118
numpy==1.23.5
numpy==1.24.3
opencv-python==4.8.1.78
onnx==1.16.0
insightface==0.7.3
@ -13,11 +13,11 @@ torch==2.0.1; sys_platform == 'darwin'
torchvision==0.15.2+cu118; sys_platform != 'darwin'
torchvision==0.15.2; sys_platform == 'darwin'
onnxruntime==1.18.0; sys_platform == 'darwin' and platform_machine != 'arm64'
onnxruntime-silicon==1.13.1; sys_platform == 'darwin' and platform_machine == 'arm64'
onnxruntime-silicon==1.16.3; sys_platform == 'darwin' and platform_machine == 'arm64'
onnxruntime-gpu==1.18.0; sys_platform != 'darwin'
tensorflow==2.13.0rc1; sys_platform == 'darwin'
tensorflow==2.12.0; sys_platform != 'darwin'
opennsfw2==0.10.2
protobuf==4.23.2
tqdm==4.66.4
gfpgan==1.3.6
gfpgan==1.3.8