using single requirements file

2023-08-06 10:32:53 -07:00
parent d1a37e49ad
commit a066a9e696
2 changed files with 5 additions and 5 deletions
--- a/frozen_requirements.txt
+++ b/frozen_requirements.txt
@@ -8,4 +8,9 @@ statsmodels==0.14.0
 lifelines==0.27.7
 pygam==0.9.0
 l0bnb==1.0.0
+torch==2.0.1
+torchvision==0.15.2
+pytorch-lightning==2.0.6
+torchinfo==1.8.0
+torchmetrics==1.0.1

--- a/torch_requirements.txt
+++ b/torch_requirements.txt
@@ -1,5 +0,0 @@
-torch==2.0.1
-torchvision==0.15.2
-pytorch-lightning==2.0.6
-torchinfo==1.8.0
-torchmetrics==1.0.1