[F] Fix torch version and rm lightning

DocsaidLab · Dec 18, 2024 · 7504daf · 7504daf
1 parent 67ecc9d
commit 7504daf
Showing 1 changed file with 3 additions and 5 deletions.
diff --git 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
+    docs
+    tests