Merge pull request #45 from openclimatefix/bug-torchvision-nms

Bug torchvision nms
openclimatefix · Feb 9, 2024 · bba679e · bba679e
2 parents 511f3d4 + 06cb854
commit bba679e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,7 +1,7 @@
 pydantic
+pytorch-lightning==2.1.3
 torch[cpu]==2.2.0
-torchvision==0.17.0
-PVNet-summation==0.1.*
+PVNet-summation==0.1.2
 pvnet==2.6.10
 ocf_datapipes==3.2.7
 nowcasting_datamodel>=1.5.30