Skip to content

Commit 9561e53

Browse files
authored
install dberta model in ci docker image (#47)
1 parent 64187ab commit 9561e53

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

docker/ci/Dockerfile

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@ RUN cd /tmp/crossfit && \
88

99
ENV CF_HOME /root/.cf
1010
RUN python3 -c "from sentence_transformers import SentenceTransformer; SentenceTransformer('all-MiniLM-L6-v2')" && \
11+
python3 -c "from transformers import AutoModel, AutoTokenizer; AutoModel.from_pretrained('microsoft/deberta-v3-base'); AutoTokenizer.from_pretrained('microsoft/deberta-v3-base')" && \
1112
cd /tmp && \
1213
mkdir -p ${CF_HOME}/raw && \
1314
wget https://public.ukp.informatik.tu-darmstadt.de/thakur/BEIR/datasets/fiqa.zip && \
@@ -18,4 +19,6 @@ RUN python3 -c "from sentence_transformers import SentenceTransformer; SentenceT
1819
rm /tmp/hotpotqa.zip && \
1920
wget https://public.ukp.informatik.tu-darmstadt.de/thakur/BEIR/datasets/nq.zip && \
2021
unzip nq.zip -d ${CF_HOME}/raw && \
21-
rm /tmp/nq.zip
22+
wget https://public.ukp.informatik.tu-darmstadt.de/thakur/BEIR/datasets/scifact.zip && \
23+
unzip scifact.zip -d ${CF_HOME}/raw && \
24+
rm /tmp/scifact.zip

0 commit comments

Comments
 (0)