diff --git a/pyproject.toml b/pyproject.toml
index a39c3f0e..5e2c2fc4 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -54,7 +54,7 @@ packages = [
 [tool.poetry.dependencies]
 python = "^3.10"
 torch = ">=2.1.1,<3.0"
-einops = "0.8.0"
+einops = "0.8.1"
 bitsandbytes = "*"
 transformers = "4.44.2"
 einops-exts = "0.0.4"