add dependencies

aistairc · Apr 11, 2024 · 37d68a1 · 37d68a1
1 parent 27930dc
commit 37d68a1
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -16,10 +16,11 @@ classifiers = [
     "License :: OSI Approved :: MIT License",
 ]
 dependencies = [
+    "colorama",
     "numpy",
-    "lightning",
+    "lightning>=2.2.1",
     "hydra-core",
-    "torch",
+    "torch>=2.2.0",
     "omegaconf",
     "h5py"
 ]