diff --git a/pyproject.toml b/pyproject.toml
index 6538b13..e2b018d 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -17,9 +17,8 @@ pytorch-lightning = "^2.0.0"
 langchain = "^0.0.334"
 hydra-core = "^1.3.2"
 tensorboard = "^2.15.1"
-lion-pytorch = "^0.1.2"
-high-order-layers-torch = "^2.4.1"
 einops = "^0.7.0"
+high-order-layers-torch = "^2.6.0"
 
 [tool.poetry.group.dev.dependencies]
 black = "^23.11.0"