This commit is contained in:
Jorg Doku
2023-08-07 11:28:36 -05:00
parent fdfe69c717
commit 72be7ff31d
2 changed files with 2 additions and 2 deletions
+1 -1
View File
@@ -31,7 +31,7 @@ RUN --mount=type=cache,target=/root/.cache/pip \
ADD src .
# Quick temporary updates
RUN pip install git+https://github.com/runpod/runpod-python@fix_scaling#egg=runpod --compile
RUN pip install git+https://github.com/runpod/runpod-python@main#egg=runpod --compile
# Prepare the models inside the docker image
ARG HUGGING_FACE_HUB_TOKEN=
+1 -1
View File
@@ -3,4 +3,4 @@
fastapi==0.99.1
vllm==0.1.2
huggingface-hub==0.16.4
runpod @ git+https://github.com/runpod/runpod-python@fix_scaling#egg=runpod
runpod @ git+https://github.com/runpod/runpod-python@main#egg=runpod