File size: 653 Bytes
13c6f75
 
4e96fe8
13c6f75
 
 
ae4a830
13c6f75
 
e97517f
 
 
13c6f75
 
 
 
 
4e96fe8
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
FROM nvidia/cuda:12.2.0-devel-ubuntu22.04

RUN useradd -m -u 1000 user

ENV PATH=/opt/conda/bin:$PATH

RUN pip install --no-cache-dir --upgrade -r requirements.txt
RUN git clone -b energy_star_dev https://github.com/huggingface/optimum-benchmark.git /optimum-benchmark && cd optimum-benchmark && pip install -e .

USER user
WORKDIR /app

# Expose the secret token at buildtime and use its value as git remote URL
RUN --mount=type=secret,id=BULK_ENERGY_TOKEN,mode=0444,required=true \
 git init && \
 git remote add origin $(cat /run/secrets/BULK_ENERGY_TOKEN)


COPY --chown=user . /app
CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]