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

Add in inswrapper_128.onnx

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

1
docker/Dockerfile.nvidia
View File

@@ -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 RUN git clone https://github.com/xaviviro/refacer && cd refacer && pip install -r requirements-GPU.txt
WORKDIR /refacer WORKDIR /refacer
COPY inswapper_128.onnx /refacer/.
# Test following commands in container to make sure GPU stuff works # Test following commands in container to make sure GPU stuff works
# nvidia-smi # nvidia-smi

6
docker/run.sh
View File

@@ -1,5 +1,11 @@
#!/bin/bash #!/bin/bash
# Run this script from within the refacer/docker folder. # 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 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 && \ docker run --rm -v $(pwd)/out:/refacer/out -p 7860:7860 --gpus all refacer python3 app.py --server_name 0.0.0.0 && \