File size: 1,643 Bytes
f980d4c
36fdd36
 
 
 
 
 
4c74ab6
36fdd36
 
 
e3571c1
36fdd36
 
 
 
 
 
 
 
f980d4c
36fdd36
f980d4c
36fdd36
48843fe
 
 
 
4cf6de5
48843fe
 
 
a916f49
 
b899ff8
a916f49
 
aa739dd
36fdd36
5de7e46
aa739dd
5de7e46
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
FROM nvidia/cuda:11.8.0-base-ubuntu22.04

# Basic installs
ARG DEBIAN_FRONTEND=noninteractive
ENV TZ='America/Detroit'
RUN apt-get update -qq \
    && apt-get -y --no-install-recommends install \
       build-essential software-properties-common wget git tar rsync ninja-build \
    && apt-get clean all \
    && rm -r /var/lib/apt/lists/*

# Install Miniconda3 23.3.1
ENV PATH="/root/.local/miniconda3/bin:$PATH"
RUN mkdir -p /root/.local \
    && wget https://repo.anaconda.com/miniconda/Miniconda3-py39_23.3.1-0-Linux-x86_64.sh \
    && mkdir /root/.conda \
    && bash Miniconda3-py39_23.3.1-0-Linux-x86_64.sh -b -p /root/.local/miniconda3 \
    && rm -f Miniconda3-py39_23.3.1-0-Linux-x86_64.sh \
    && ln -sf /root/.local/miniconda3/etc/profile.d/conda.sh /etc/profile.d/conda.sh

# Install spitfight
ADD . /workspace/leaderboard
RUN cd /workspace/leaderboard && pip install -e .[benchmark]

# Clone lm-evaluation-harness and install
RUN cd /workspace \
      && git clone https://github.com/EleutherAI/lm-evaluation-harness.git \
      && cd lm-evaluation-harness \
      && git checkout d1537059b515511801ae9b742f8e949f1bfcd010 \
      && rm -r .git \
      && pip install -e .

# Apply patches
# Salesforce xgen inference fix (https://github.com/lm-sys/FastChat/pull/2350)
RUN cd /root/.local/miniconda3/lib/python3.9/site-packages \
      && patch -p1 < /workspace/leaderboard/deployment/fastchat_xgen_fix.patch

# Where all the weights downloaded from Hugging Face Hub will go to
ENV TRANSFORMERS_CACHE=/data/leaderboard/hfcache

# So that docker exec container python scripts/benchmark.py will work
WORKDIR /workspace/leaderboard