From 4dc47462357d52d205a65cc9dd77819a74789811 Mon Sep 17 00:00:00 2001 From: Kenneth Estanislao Date: Wed, 21 Aug 2024 14:40:15 +0800 Subject: [PATCH] update inswapper --- README.md | 2 +- modules/processors/frame/face_swapper.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index fe60fe5..361e6f5 100644 --- a/README.md +++ b/README.md @@ -24,7 +24,7 @@ Users of this software are expected to use this software responsibly while abidi #### 3. Download Models 1. [GFPGANv1.4](https://huggingface.co/hacksider/deep-live-cam/resolve/main/GFPGANv1.4.pth) - 2. [inswapper_128_fp16.onnx](https://huggingface.co/hacksider/deep-live-cam/resolve/main/inswapper_128_fp16.onnx) + 2. [inswapper_128_fp16.onnx](https://huggingface.co/hacksider/deep-live-cam/resolve/main/inswapper_128.onnx) Then put those 2 files on the "**models**" folder diff --git a/modules/processors/frame/face_swapper.py b/modules/processors/frame/face_swapper.py index 7bf5550..6a228c5 100644 --- a/modules/processors/frame/face_swapper.py +++ b/modules/processors/frame/face_swapper.py @@ -18,7 +18,7 @@ NAME = 'DLC.FACE-SWAPPER' def pre_check() -> bool: download_directory_path = resolve_relative_path('../models') conditional_download(download_directory_path, [ - 'https://huggingface.co/hacksider/deep-live-cam/blob/main/inswapper_128_fp16.onnx' + 'https://huggingface.co/hacksider/deep-live-cam/blob/main/inswapper_128.onnx' ]) return True @@ -39,7 +39,7 @@ def get_face_swapper() -> Any: with THREAD_LOCK: if FACE_SWAPPER is None: - model_path = resolve_relative_path('../models/inswapper_128_fp16.onnx') + model_path = resolve_relative_path('../models/inswapper_128.onnx') FACE_SWAPPER = insightface.model_zoo.get_model(model_path, providers=modules.globals.execution_providers) return FACE_SWAPPER