diff --git a/pyproject.toml b/pyproject.toml index 5392650..27822e7 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -66,7 +66,7 @@ dependencies = [ "torch>=2.5.1", "torchvision>=0.20.1", "tqdm==4.66.5", - "transformers==4.21.1", + "transformers==4.53.0", "vector-quantize-pytorch==1.20.11", "wandb==0.13.1", ] diff --git a/requirements.txt b/requirements.txt index 6d4ebc8..a38b1f8 100644 --- a/requirements.txt +++ b/requirements.txt @@ -31,6 +31,6 @@ timm==1.0.8 torch>=2.5.1 torchvision>=0.20.1 tqdm==4.66.5 -transformers==4.21.1 +transformers==4.53.0 vector-quantize-pytorch==1.20.11 wandb==0.13.1