From 7222d9aa47bd7dbeddf563956d0f9a9807f174de Mon Sep 17 00:00:00 2001 From: inisis Date: Mon, 12 Aug 2024 02:25:55 +0000 Subject: [PATCH] add onnxslim support Signed-off-by: inisis --- README.md | 5 ++++- requirements.txt | 1 + 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index d8e004f..c81f33d 100644 --- a/README.md +++ b/README.md @@ -26,7 +26,10 @@ Users of this software are expected to use this software responsibly while abidi 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) -Then put those 2 files on the "**models**" folder +Then put those 2 files on the "**models**" folder, you can further optimize inswapper_128_fp16.onnx by using +``` +onnxslim inswapper_128_fp16.onnx inswapper_128_fp16.onnx +``` #### 4. Install dependency We highly recommend to work with a `venv` to avoid issues. diff --git a/requirements.txt b/requirements.txt index f65195e..0beb7a6 100644 --- a/requirements.txt +++ b/requirements.txt @@ -21,3 +21,4 @@ opennsfw2==0.10.2 protobuf==4.23.2 tqdm==4.66.4 gfpgan==1.3.8 +onnxslim