diff --git a/Dockerfile b/Dockerfile index bf5d59e..8234395 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 29943bf..13f0e27 100644 --- a/setup.py +++ b/setup.py @@ -28,7 +28,8 @@ setuptools.setup( 'scipy', 'librosa', 'transformers==4.31.0', - 'tokenizers', + 'tokenizers==0.13.4.rc3', + 'scipy==1.13.1' # 'deepspeed==0.8.3', ], classifiers=[