diff --git a/requirements.txt b/requirements.txt index 74b0e04..ea8ef56 100644 --- a/requirements.txt +++ b/requirements.txt @@ -6,7 +6,7 @@ imageio==2.34.1 matplotlib==3.9.0 numpy==1.26.4 positional-encodings==6.0.3 -torch==2.4.0 +torch==2.7.0 tqdm==4.66.4 vector-quantize-pytorch==1.14.24 wandb==0.17.0