Merge pull request #809 from nikitalokhmachev-ai/docker-fix

Fixing issues with Docker Setup
neonbjb · Aug 13, 2024 · 2d6d386 · 2d6d386
2 parents 86100b6 + ef5e8e2
commit 2d6d386
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git 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
@@ -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=[