diff --git a/src/mloq/assets/requirements/pytorch.txt b/src/mloq/assets/requirements/pytorch.txt index ce0b46de..667684e7 100644 --- a/src/mloq/assets/requirements/pytorch.txt +++ b/src/mloq/assets/requirements/pytorch.txt @@ -1,4 +1,4 @@ -torch==1.10.0 +torch==2.2.0 torchvision==0.11.1 einops==0.3.2 pytorch-lightning==1.5.2 \ No newline at end of file