diff --git a/requirements.txt b/requirements.txt index b110655..1c66421 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,6 @@ huggingface-hub==0.18.0 timm==0.9.7 -torch==2.1.0+cu121 +torch==2.2.0 torchmetrics==1.2.0 torchvision==0.16.0+cu121 tqdm==4.66.1