diff --git a/poetry.lock b/poetry.lock
index 3ac2df37..24cfd6c3 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -2323,4 +2323,4 @@ multidict = ">=4.0"
 [metadata]
 lock-version = "2.0"
 python-versions = "~3.11"
-content-hash = "1628bb8aa37d6bbc0d9b70f05cf2acd5743af96cc8359491ffa29fc530fbfab5"
+content-hash = "a213726efbd045e22548ffee62fc7bd92b15755d4925e04671ab4c9e91acb7e1"
diff --git a/pyproject.toml b/pyproject.toml
index 48c459a2..470175c2 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -13,7 +13,7 @@ pandas = "^2.2.2"
 matplotlib = "^3.9.1"
 migrate-ckpt = {git = "https://github.com/bdvllrs/migrate-ckpt.git", rev = "v0.2.0"}
 click = "^8.1.7"
-lightning = "^2.3.3"
+lightning = ">=2.1.0"
 torch = "^2.0.1"
 
 [tool.poetry.group.dev.dependencies]