Fix dependencies

will-rice · Feb 24, 2025 · 22e3782 · 22e3782
1 parent 7c5414a
commit 22e3782
Show file tree

Hide file tree

Showing 2 changed files with 535 additions and 13 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -37,7 +37,7 @@ dependencies = [
     "torchaudio>=2.5.1",
     "transformers>=4.46.3",
     "torchvision>=0.20.1",
-    "torchmetrics>=1.6.1",
+    "pytorch-lightning>=2.5.0.post0",
 ]
 
 [project.urls]