diff --git a/docker/requirements.txt b/docker/requirements.txt index a2856d6..b0ab5d5 100644 --- a/docker/requirements.txt +++ b/docker/requirements.txt @@ -29,6 +29,6 @@ timm torch==2.3.1 --index-url https://download.pytorch.org/whl/cu121 torchvision==0.18.1 --index-url https://download.pytorch.org/whl/cu121 torchaudio==2.3.1 --index-url https://download.pytorch.org/whl/cu121 -transformers==4.40.1 +transformers==4.53.0 wandb xformers==0.0.27