Spaces:
Running
on
A100
Running
on
A100
dockerfile
Browse files- Dockerfile +30 -0
Dockerfile
ADDED
@@ -0,0 +1,30 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
FROM python:3.11.9-slim-bullseye
|
2 |
+
|
3 |
+
USER root
|
4 |
+
|
5 |
+
RUN adduser --quiet --disabled-password \
|
6 |
+
--home /home/conv_user \
|
7 |
+
--shell /bin/bash conv_user
|
8 |
+
RUN adduser conv_user sudo
|
9 |
+
|
10 |
+
WORKDIR /srv
|
11 |
+
ENV PYTHONPATH="/srv"
|
12 |
+
|
13 |
+
RUN apt-get update && apt-get install -y git && rm -rf /var/lib/apt/lists/*
|
14 |
+
|
15 |
+
RUN apt-get -y update
|
16 |
+
RUN apt-get install -y --no-install-recommends \
|
17 |
+
build-essential \
|
18 |
+
python3-dev \
|
19 |
+
libglib2.0-0 \
|
20 |
+
libgl1-mesa-glx \
|
21 |
+
libpq-dev \
|
22 |
+
make
|
23 |
+
# insall latest vllm
|
24 |
+
RUN pip install vllm==0.7.2
|
25 |
+
|
26 |
+
# Install transformers from source, need specific revision for Qwen2.5 as
|
27 |
+
# breaking changes in master
|
28 |
+
RUN pip install --upgrade git+https://github.com/huggingface/transformers.git@336dc69d63d56f232a183a3e7f52790429b871ef
|
29 |
+
|
30 |
+
ENTRYPOINT ["sh", "-c", "python -m vllm.entrypoints.openai.api_server --model convergence-ai/proxy-lite --trust-remote-code --tokenizer-pool-size 10 --limit-mm-per-prompt image=2 --enable-auto-tool-choice --tool-call-parser hermes --port 7860"]
|