Tomtom84 commited on
Commit
c67da96
·
verified ·
1 Parent(s): 97aa62e

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -34,7 +34,7 @@ COPY --chown=user . /app
34
 
35
  RUN pip install --upgrade pip
36
  RUN pip install --no-cache-dir -r requirements.txt
37
- RUN CMAKE_ARGS="-DGGML_CUDA=on" pip install outetts --upgrade
38
 
39
  #CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860", , "--ws", "auto", "--allow-websocket-origin", "*"]
40
  CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
 
34
 
35
  RUN pip install --upgrade pip
36
  RUN pip install --no-cache-dir -r requirements.txt
37
+ RUN ENV CMAKE_ARGS="-DGGML_CUDA=on -DCMAKE_CUDA_ARCHITECTURES=86" pip install outetts --upgrade
38
 
39
  #CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860", , "--ws", "auto", "--allow-websocket-origin", "*"]
40
  CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]