From 0b64095e4659ef3ac5971ef6b0c1e172206df169 Mon Sep 17 00:00:00 2001 From: Andras Fekete Date: Fri, 16 Jun 2023 09:53:22 -0400 Subject: [PATCH] Better to use the hosts' version of refacer --- docker/Dockerfile.nvidia | 1 - docker/run.sh | 5 +++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/docker/Dockerfile.nvidia b/docker/Dockerfile.nvidia index 9b52b28..407d221 100644 --- a/docker/Dockerfile.nvidia +++ b/docker/Dockerfile.nvidia @@ -14,7 +14,6 @@ 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 a20046a..f4a4f04 100755 --- a/docker/run.sh +++ b/docker/run.sh @@ -7,6 +7,7 @@ # or https://1drv.ms/u/s!AsHA3Xbnj6uAgxhb_tmQ7egHACOR?e=CPoThO # or https://civitai.com/models/80324?modelVersionId=85159 +docker stop -t 0 refacer 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 && \ - sleep 1 && google-chrome --new-window "http://127.0.0.1:7860" & + docker run --rm --name refacer -v $(pwd)/..:/refacer -p 7860:7860 --gpus all refacer python3 app.py --server_name 0.0.0.0 & +sleep 2 && google-chrome --new-window "http://127.0.0.1:7860" &