diff --git a/server/Dockerfile b/server/Dockerfile index 0a9ec86..ea40524 100644 --- a/server/Dockerfile +++ b/server/Dockerfile @@ -6,6 +6,6 @@ RUN python -m pip install --no-deps -r requirements.frozen.txt \ COPY main.py . -ENV NUM_THREADS=8 +ENV NUM_THREADS=2 EXPOSE 80 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "80"] diff --git a/server/Dockerfile.cuda118 b/server/Dockerfile.cuda118 index d8fb1a1..d3eb115 100644 --- a/server/Dockerfile.cuda118 +++ b/server/Dockerfile.cuda118 @@ -9,6 +9,6 @@ COPY main.py . #Mark this 1 if you have older card ENV NVIDIA_DISABLE_REQUIRE=0 -ENV NUM_THREADS=8 +ENV NUM_THREADS=2 EXPOSE 80 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "80"]