diff --git a/docker/Dockerfile.nvidia b/docker/Dockerfile.nvidia index 407d221..9b52b28 100644 --- a/docker/Dockerfile.nvidia +++ b/docker/Dockerfile.nvidia @@ -14,6 +14,7 @@ RUN pip install nvidia-tensorrt RUN git clone https://github.com/xaviviro/refacer && cd refacer && pip install -r requirements-GPU.txt WORKDIR /refacer +COPY inswapper_128.onnx /refacer/. # Test following commands in container to make sure GPU stuff works # nvidia-smi diff --git a/docker/run.sh b/docker/run.sh index e18f6cf..a20046a 100755 --- a/docker/run.sh +++ b/docker/run.sh @@ -1,5 +1,11 @@ #!/bin/bash # Run this script from within the refacer/docker folder. +# You'll need inswrapper_128.onnx from either: +# https://drive.google.com/file/d/1eu60OrRtn4WhKrzM4mQv4F3rIuyUXqfl/view?usp=drive_link +# or https://drive.google.com/file/d/1jbDUGrADco9A1MutWjO6d_1dwizh9w9P/view?usp=sharing +# or https://mega.nz/file/9l8mGDJA#FnPxHwpdhDovDo6OvbQjhHd2nDAk8_iVEgo3mpHLG6U +# or https://1drv.ms/u/s!AsHA3Xbnj6uAgxhb_tmQ7egHACOR?e=CPoThO +# or https://civitai.com/models/80324?modelVersionId=85159 docker build -t refacer -f Dockerfile.nvidia . && \ docker run --rm -v $(pwd)/out:/refacer/out -p 7860:7860 --gpus all refacer python3 app.py --server_name 0.0.0.0 && \