tgi-bench / Dockerfile
hlarcher's picture
hlarcher HF Staff
Update Dockerfile
6dac797 verified
raw
history blame contribute delete
813 Bytes
FROM python:3-bookworm
RUN apt-get update && apt-get install -y \
git \
&& rm -rf /var/lib/apt/lists/* \
ENV POETRY_HOME=/poetry
RUN mkdir /poetry && curl -sSL https://install.python-poetry.org | POETRY_HOME=/poetry python3 -
ENV PATH="/poetry/bin/:$PATH"
RUN poetry config virtualenvs.create true \
&& poetry config cache-dir /poetry/cache \
&& poetry config installer.parallel false \
&& poetry config virtualenvs.in-project true
WORKDIR /app
CMD ["bash"]
RUN git clone https://github.com/huggingface/inference-benchmarker.git \
&& cd inference-benchmarker \
&& poetry install --no-root
COPY results results
WORKDIR /app/inference-benchmarker
RUN chmod 777 -R /app/inference-benchmarker
CMD ["poetry", "run", "python", "extra/dashboard/app.py", "--from-results-dir", "/app/results"]