Merge pull request #9 from SkywardAI/release/new

Split ML libs from Kirin

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

requirements.txt

@@ -0,0 +1,3 @@
+torch == 2.2.2
+transformers == 4.39.2
+sentencepiece == 0.2.0