diff --git a/docs/Project.toml b/docs/Project.toml
index 55bcfb6a7..e13659698 100644
--- a/docs/Project.toml
+++ b/docs/Project.toml
@@ -25,7 +25,7 @@ DomainSets = "0.6"
 Flux = "0.13, 0.14"
 Integrals = "3.3"
 IntegralsCubature = "0.2"
-Lux = "0.4"
+Lux = "0.4, 0.5"
 ModelingToolkit = "8.33"
 NeuralPDE = "5.3"
 Optimization = "3.9"