diff --git a/requirements.txt b/requirements.txt index 941629b..6e13c02 100644 --- a/requirements.txt +++ b/requirements.txt @@ -109,7 +109,7 @@ terminado==0.9.1 testpath==0.4.4 threadpoolctl==2.1.0 toml==0.10.1 -torch==1.6.0 +torch==2.6.0 torchvision==0.7.0 tornado==6.0.4 tqdm==4.50.0