diff --git a/setup.cfg b/setup.cfg
index 85db3c57..117062a8 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -17,8 +17,9 @@ install_requires =
     huggingface_hub>=0.10
     gymnasium<=1.0.0
     stable-baselines3>=2.0.0,<=2.4.0
+    torch<=2.8.0
     huggingface_sb3
-    onnx
+    onnx<=1.19.1
     onnxruntime
 
 python_requires = >=3.8