diff --git a/requirements.txt b/requirements.txt index 5c21fc3..23c29c5 100644 --- a/requirements.txt +++ b/requirements.txt @@ -22,6 +22,6 @@ scikit-learn==1.5.0 scipy==1.1.0 six==1.11.0 toolz==0.9.0 -torch==2.2.0 +torch==2.6.0 torchvision==0.2.1 tqdm==4.24.0