Skip to content

Commit a6b8d05

Browse files
committed
fixed docker
1 parent 158c2aa commit a6b8d05

File tree

2 files changed

+7
-3
lines changed

2 files changed

+7
-3
lines changed

Dockerfile

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,10 +26,12 @@ ENV HNSWLIB_NO_NATIVE=1
2626
RUN apt-get install -y python3-dev
2727
RUN apt-get install -y build-essential
2828

29+
RUN pip install uv
30+
2931
# Install CPU-only pytorch, eliminating huge nvidia dependencies
3032
RUN pip install torch==2.3.1+cpu -f https://download.pytorch.org/whl/torch_stable.html
3133
RUN pip install https://github.com/pepkit/pipestat/archive/refs/heads/dev.zip
3234

33-
RUN pip install -r requirements/requirements-all.txt --no-cache-dir
35+
RUN uv pip install -r requirements/requirements-all.txt --no-cache-dir --system
3436

3537
CMD ["uvicorn", "bedhost.main:app", "--host", "0.0.0.0", "--port", "80"]

dev.Dockerfile

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -26,10 +26,12 @@ ENV HNSWLIB_NO_NATIVE=1
2626
RUN apt-get install -y python3-dev
2727
RUN apt-get install -y build-essential
2828

29+
RUN pip install uv
30+
2931
# Install CPU-only pytorch, eliminating huge nvidia dependencies
3032
RUN pip install torch==2.3.1+cpu -f https://download.pytorch.org/whl/torch_stable.html
3133
RUN pip install https://github.com/pepkit/pipestat/archive/refs/heads/dev.zip
3234

33-
RUN pip install -r requirements/requirements-all.txt --no-cache-dir
35+
RUN uv pip install -r requirements/requirements-all.txt --no-cache-dir --system
3436

35-
CMD ["uvicorn", "bedhost.main:app", "--host", "0.0.0.0", "--port", "80"]
37+
CMD ["uvicorn", "bedhost.main:app", "--host", "0.0.0.0", "--port", "80"]

0 commit comments

Comments
 (0)