diff --git a/Dockerfile b/Dockerfile index db87afa..8db317b 100644 --- a/Dockerfile +++ b/Dockerfile @@ -29,6 +29,6 @@ SHELL ["/bin/bash", "--login", "-c"] RUN conda create --name tortoise python=3.9 numba inflect \ && conda activate tortoise \ && conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia \ - && conda install transformers=4.29.2 \ + && conda install transformers=4.31.0 \ && cd /app \ && python setup.py install