dream

1
# syntax=docker/dockerfile:experimental
2

3
FROM pytorch/pytorch:1.5-cuda10.1-cudnn7-runtime
4

5
RUN : \
6
	&& apt-get update \
7
	&& DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
8
	software-properties-common \
9
	&& add-apt-repository -y ppa:deadsnakes/ppa \
10
	&& DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends python3.10\
11
	&& apt-get clean \
12
	&& rm -rf /var/lib/apt/lists/* \
13
	&& :
14

15
RUN apt-get update && \
16
	apt-get upgrade -y && \
17
	apt-get install -y git libsqlite3-dev
18

19
RUN python3 -m venv /venv
20
ENV PATH=/venv/bin:$PATH
21
RUN python3 --version
22
WORKDIR /src
23

24
ARG SERVICE_PORT
25
ENV SERVICE_PORT ${SERVICE_PORT}
26
ARG PRETRAINED_MODEL_NAME_OR_PATH
27
ENV PRETRAINED_MODEL_NAME_OR_PATH ${PRETRAINED_MODEL_NAME_OR_PATH}
28
ARG PAIR_DIALOG_HISTORY_LENGTH
29
ENV PAIR_DIALOG_HISTORY_LENGTH ${PAIR_DIALOG_HISTORY_LENGTH}
30
ARG CHAT_EVERY_SENT_MAX_LENGTH
31
ENV CHAT_EVERY_SENT_MAX_LENGTH ${CHAT_EVERY_SENT_MAX_LENGTH}
32
ARG PERSONA_EVERY_SENT_MAX_LENGTH
33
ENV PERSONA_EVERY_SENT_MAX_LENGTH ${PERSONA_EVERY_SENT_MAX_LENGTH}
34
ARG GENERATION_PARAMS_CONFIG
35
ENV GENERATION_PARAMS_CONFIG ${GENERATION_PARAMS_CONFIG}
36

37
COPY ./services/seq2seq_persona_based/requirements.txt /src/requirements.txt
38
# RUN --mount=type=cache,target=/root/.cache pip install -r /src/requirements.txt
39
RUN pip install --upgrade pip && pip install -r /src/requirements.txt
40

41
COPY ./common/ /src/common/
42

43
COPY ./services/seq2seq_persona_based/ /src
44

45
CMD gunicorn --workers=1 server:app -b 0.0.0.0:${SERVICE_PORT} --timeout=300
46