From 7817d60bc4030f19217faeb9c01511bbd291456d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Alexandre=20Z=C3=BCrcher?= Date: Sat, 24 Feb 2024 14:06:28 +0200 Subject: [PATCH] Dockerfile - updated transformers version to 4.31.0 --- Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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