From 7504dafa2840285fa803c616f148da67f7fd6e7b Mon Sep 17 00:00:00 2001
From: Jack Lin <p12922005@ntu.edu.tw>
Date: Wed, 18 Dec 2024 20:37:01 +0800
Subject: [PATCH] [F] Fix torch version and rm lightning

---
 setup.cfg | 8 +++-----
 1 file changed, 3 insertions(+), 5 deletions(-)

diff --git a/setup.cfg b/setup.cfg
index 00cdde7..a0dbf76 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -29,18 +29,16 @@ install_requires =
     timm>=0.5.4
     scikit-learn
     transformers
-    torch
+    torch>=2.4.0
     torchvision
     torchmetrics
     albumentations
     ptflops==0.7.0
     calflops
-    lightning
     rich
 
 [options.packages.find]
 exclude =
     docker
-    demo
-    tests
-    benchmarks
\ No newline at end of file
+    docs
+    tests
\ No newline at end of file