update litgpt

whittle-org · Jul 24, 2024 · 19ec01f · 19ec01f
1 parent 891747f
commit 19ec01f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -14,7 +14,7 @@ matplotlib = "^3.9.0"
 transformers = "^4.42.3"
 typing-extensions = "^4.12.2"
 torch = ">=2.2.0"
-litgpt = { extras = ["all"], version = ">=0.4.2" }
+litgpt = { extras = ["all"], version = "0.4.6" }
 syne-tune = { extras = ["moo"], version = "^0.13.0" }
 torchvision = "^0.18.1"