From e48a4856dfc09025f541a79478218d7962ce4f21 Mon Sep 17 00:00:00 2001 From: Nathan Lambert Date: Fri, 6 Sep 2024 00:53:28 +0000 Subject: [PATCH] fix vllm version --- Dockerfile | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/Dockerfile b/Dockerfile index 3afd9c1..5ecb3f9 100644 --- a/Dockerfile +++ b/Dockerfile @@ -56,7 +56,7 @@ COPY scripts scripts COPY setup.py setup.py COPY Makefile Makefile COPY README.md README.md -RUN pip install -e . +RUN pip install -e .[generative] RUN chmod +x scripts/* # this is just very slow @@ -74,9 +74,5 @@ RUN pip install openai RUN pip install together RUN pip install google-generativeai -# updated for Gemma 2 -RUN pip install vllm==0.5.1 -# from git+https://github.com/vllm-project/vllm.git@d87f39e9a9dd149f5dd7a58b4d98b21f713827b6 - # for interactive session RUN chmod -R 777 /stage/