From 53921a69c9e6932f4e5098fab971a04260815446 Mon Sep 17 00:00:00 2001 From: Aisuko Date: Mon, 1 Apr 2024 03:31:45 +0000 Subject: [PATCH] Splt ML to backend Signed-off-by: GitHub --- Dockerfile | 6 ++++++ requirements.txt | 3 +++ 2 files changed, 9 insertions(+) create mode 100644 requirements.txt diff --git a/Dockerfile b/Dockerfile index 85d210a..4ec78ab 100644 --- a/Dockerfile +++ b/Dockerfile @@ -12,4 +12,10 @@ ENV PYTHONDONTWRITEBYTECODE 1 # the stdout and stderr streams are sent straight to terminal (e.g. your container log) without being first buffered and that you can see the output of your application (e.g. django logs) in real tim ENV PYTHONUNBUFFERED 1 +# Insalling ML dependencies inside requirements.txt +COPY requirements.txt /tmp/pip-tmp/ +RUN pip --disable-pip-version-check --no-cache-dir install -r /tmp/pip-tmp/requirements.txt \ + && rm -rf /tmp/pip-tmp + + RUN pip --disable-pip-version-check --no-cache-dir install -r https://raw.githubusercontent.com/SkywardAI/chat-backend/main/backend/requirements.txt diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..7e43509 --- /dev/null +++ b/requirements.txt @@ -0,0 +1,3 @@ +torch == 2.2.2 +transformers == 4.39.2 +sentencepiece == 0.2.0