diff --git a/requirements.txt b/requirements.txt index c3cb775..1246d5e 100644 --- a/requirements.txt +++ b/requirements.txt @@ -9,7 +9,7 @@ shortuuid==1.0.11 tensorboard==2.10.0 tensorboard-data-server==0.6.1 tensorboard-plugin-wit==1.8.1 -torch==1.12.1 +torch==2.9.1 torchaudio==0.12.1 torchmetrics==0.9.3 torchvision==0.13.1