diff --git a/requirements.txt b/requirements.txt index 5d286375..4f59824e 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,6 @@ lightning-cloud == 0.5.64 # Must be pinned to ensure compatibility lightning-utilities >=0.8.0, <0.10.0 -torch >=2.1.0, <=2.2.0 +torch >=2.1.0 filelock tqdm numpy