From ef5e8e2c52885e4793ba986680f1c77362cc3c65 Mon Sep 17 00:00:00 2001
From: nikitalokhmachev-ai <n1kkqt@gmail.com>
Date: Mon, 29 Jul 2024 22:41:46 -0400
Subject: [PATCH] fix: docker flow

---
 Dockerfile | 6 +++---
 setup.py   | 3 ++-
 2 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index bf5d59e2..82343952 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -26,9 +26,9 @@ RUN wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -
 # --login option used to source bashrc (thus activating conda env) at every RUN statement
 SHELL ["/bin/bash", "--login", "-c"]
 
-RUN conda create --name tortoise python=3.9 numba inflect \
+RUN conda create --name tortoise python=3.9 numba inflect -y \
     && conda activate tortoise \
-    && conda install pytorch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 pytorch-cuda=12.1 -c pytorch -c nvidia \
-    && conda install transformers=4.31.0 \
+    && conda install --yes pytorch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 pytorch-cuda=12.1 -c pytorch -c nvidia \
+    && conda install --yes transformers=4.31.0 \
     && cd /app \
     && python setup.py install
diff --git a/setup.py b/setup.py
index 29943bf3..13f0e276 100644
--- a/setup.py
+++ b/setup.py
@@ -28,7 +28,8 @@
         'scipy',
         'librosa',
         'transformers==4.31.0',
-        'tokenizers',
+        'tokenizers==0.13.4.rc3',
+        'scipy==1.13.1'
         # 'deepspeed==0.8.3',
     ],
     classifiers=[