add onnxslim support

Signed-off-by: inisis <desmond.yao@buaa.edu.cn>
2024-08-12 02:25:55 +00:00 · 2024-08-12 02:25:55 +00:00 · 7222d9aa47
parent a31e81fa66
commit 7222d9aa47
2 changed files with 5 additions and 1 deletions
--- 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.
--- 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