From a066a9e696659a276a1bf1d87afd6e32ed42de26 Mon Sep 17 00:00:00 2001 From: Jonathan Taylor Date: Sun, 6 Aug 2023 10:32:53 -0700 Subject: [PATCH] using single requirements file --- frozen_requirements.txt => requirements.txt | 5 +++++ torch_requirements.txt | 5 ----- 2 files changed, 5 insertions(+), 5 deletions(-) rename frozen_requirements.txt => requirements.txt (57%) delete mode 100644 torch_requirements.txt diff --git a/frozen_requirements.txt b/requirements.txt similarity index 57% rename from frozen_requirements.txt rename to requirements.txt index 4bc29f2..270dc42 100644 --- a/frozen_requirements.txt +++ b/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 diff --git a/torch_requirements.txt b/torch_requirements.txt deleted file mode 100644 index c6ad504..0000000 --- a/torch_requirements.txt +++ /dev/null @@ -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