diff --git a/requirements/base.txt b/requirements/base.txt index 1ae7bb7d5f..1c6c945923 100644 --- a/requirements/base.txt +++ b/requirements/base.txt @@ -2,7 +2,7 @@ packaging <24.0 setuptools <=59.5.0 # Prevent install bug with tensorboard -numpy <1.25 +numpy <1.26 torch >1.7.0, <=2.0.1 torchmetrics >0.7.0, <0.11.0 # strict pytorch-lightning >1.8.0, <2.0.0 # strict