Skip to content

Commit

Permalink
pin mistral and install adapter from branch 🚀
Browse files Browse the repository at this point in the history
Signed-off-by: Jefferson Fialho <[email protected]>
  • Loading branch information
fialhocoelho committed Nov 23, 2024
1 parent 565cfcc commit 049cf55
Show file tree
Hide file tree
Showing 4 changed files with 9 additions and 5 deletions.
8 changes: 6 additions & 2 deletions Dockerfile.ubi
Original file line number Diff line number Diff line change
Expand Up @@ -200,14 +200,18 @@ COPY examples/*.jinja /app/data/template/
USER 2000
ENTRYPOINT ["python3", "-m", "vllm.entrypoints.openai.api_server"]


FROM vllm-openai as vllm-grpc-adapter

USER root

RUN --mount=type=cache,target=/root/.cache/pip \
--mount=type=bind,from=build,src=/workspace/dist,target=/workspace/dist \
uv pip install $(echo dist/*.whl)'[tensorizer]' vllm-tgis-adapter==0.5.3
uv pip install $(echo /workspace/dist/*.whl)'[tensorizer]' --verbose && \
uv pip install \
"git+https://github.com/opendatahub-io/vllm-tgis-adapter@ibm-20241106-adapter" --verbose

RUN --mount=type=bind,from=build,src=/workspace/dist,target=/workspace/dist \
echo "Local dir and dist:" && pwd && ls -l /workspace/dist

ENV GRPC_PORT=8033 \
PORT=8000 \
Expand Down
2 changes: 1 addition & 1 deletion docs/requirements-docs.txt
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@ pydantic >= 2.8
torch
py-cpuinfo
transformers
mistral_common >= 1.3.4
mistral_common >= 1.5.0
aiohttp
starlette
openai # Required by docs/source/serving/openai_compatible_server.md's vllm.entrypoints.openai.cli_args
Expand Down
2 changes: 1 addition & 1 deletion requirements-test.in
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@ timm # required for internvl test
torch==2.5.1
transformers_stream_generator # required for qwen-vl test
matplotlib # required for qwen-vl test
mistral_common[opencv] >= 1.4.4 # required for pixtral test
mistral_common[opencv] >= 1.5.0 # required for pixtral test
datamodel_code_generator # required for minicpm3 test
lm-eval[api]==0.4.4 # required for model evaluation test

Expand Down
2 changes: 1 addition & 1 deletion requirements-test.txt
Original file line number Diff line number Diff line change
Expand Up @@ -217,7 +217,7 @@ mbstrdecoder==1.1.3
# dataproperty
# pytablewriter
# typepy
mistral-common[opencv]==1.4.4
mistral_common[opencv]==1.5.0
# via
# -r requirements-test.in
# mistral-common
Expand Down

0 comments on commit 049cf55

Please sign in to comment.