admin/LC_NeoRefacer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Better to use the hosts' version of refacer

Browse Source
This commit is contained in:
Andras Fekete
2023-06-16 09:53:22 -04:00
parent fdae43c0a1
commit 0b64095e46
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
docker/Dockerfile.nvidia
View File

@@ -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

5
docker/run.sh
View File

@@ -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" &