Dockerfile - updated transformers version to 4.31.0

This commit is contained in:
Alexandre Zürcher 2024-02-24 14:06:28 +02:00
parent 1839a380d6
commit 7817d60bc4

View file

@ -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