Add in inswrapper_128.onnx

2023-06-16 09:44:12 -04:00
parent fabee094d0
commit fdae43c0a1
2 changed files with 7 additions and 0 deletions
--- 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