diff --git a/requirements.txt b/requirements.txt index 1c9c4c7..843317c 100644 --- a/requirements.txt +++ b/requirements.txt @@ -4,7 +4,7 @@ pandas>=3.0,<4 numpy>=2.1 prophet>=1.3 scikit-learn>=1.8 -torch>=2.10,<2.11 # upper cap: torch 2.11 + pandas 3 segfault in XGBoost training — see issue #29 +torch>=2.12.0+cpu,<2.13 # upper cap: torch 2.11 + pandas 3 segfault in XGBoost training — see issue #29 # tensorflow>=2.15.0 lightgbm>=4.6 xgboost>=3.2