fixed

2023-08-06 14:17:53 -05:00
parent 401a4637fa
commit 4e4d3a27fc
2 changed files with 2 additions and 2 deletions
@@ -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@async_gen_test#egg=runpod --compile
+RUN pip install git+https://github.com/runpod/runpod-python@fix_scaling#egg=runpod --compile

 # Prepare the models inside the docker image
 ARG HUGGING_FACE_HUB_TOKEN=
@@ -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@async_gen_test#egg=runpod 
+runpod @ git+https://github.com/runpod/runpod-python@fix_scaling#egg=runpod