diff --git a/tools/tdgpt/requirements_chronos.txt b/tools/tdgpt/requirements_chronos.txt index 69e095d11eb0..306e1ad5a940 100644 --- a/tools/tdgpt/requirements_chronos.txt +++ b/tools/tdgpt/requirements_chronos.txt @@ -1,4 +1,4 @@ -f https://download.pytorch.org/whl/torch_stable.html -torch==2.6.0+cpu +torch==2.12.1 chronos-forecasting flask diff --git a/tools/tdgpt/requirements_ess.txt b/tools/tdgpt/requirements_ess.txt index 24212bc0d32f..c7a19d5def7e 100644 --- a/tools/tdgpt/requirements_ess.txt +++ b/tools/tdgpt/requirements_ess.txt @@ -2,7 +2,7 @@ transformers==4.51.0 --find-links https://download.pytorch.org/whl/torch/ -torch==2.6.0+cpu +torch==2.12.1 accelerate==0.34.2 cookiecutter>=2.1.1 ray>=2.9.0 diff --git a/tools/tdgpt/requirements_moirai.txt b/tools/tdgpt/requirements_moirai.txt index f3a5979b7125..4460e8619cee 100644 --- a/tools/tdgpt/requirements_moirai.txt +++ b/tools/tdgpt/requirements_moirai.txt @@ -1,4 +1,4 @@ -f https://download.pytorch.org/whl/torch_stable.html -torch==2.6.0+cpu +torch==2.12.1 uni2ts flask diff --git a/tools/tdgpt/requirements_moment.txt b/tools/tdgpt/requirements_moment.txt index 5840f68c922a..f48363fce962 100644 --- a/tools/tdgpt/requirements_moment.txt +++ b/tools/tdgpt/requirements_moment.txt @@ -1,5 +1,5 @@ -f https://download.pytorch.org/whl/torch_stable.html -torch==2.6.0+cpu +torch==2.12.1 transformers==4.51.0 numpy==1.25.2 matplotlib diff --git a/tools/tdgpt/requirements_timesfm.txt b/tools/tdgpt/requirements_timesfm.txt index a63a8a967ac9..7b4d496d605e 100644 --- a/tools/tdgpt/requirements_timesfm.txt +++ b/tools/tdgpt/requirements_timesfm.txt @@ -1,5 +1,5 @@ -f https://download.pytorch.org/whl/torch_stable.html -torch==2.6.0+cpu +torch==2.12.1 jax timesfm flask==3.0.3 diff --git a/tools/tdgpt/requirements_windows_core.txt b/tools/tdgpt/requirements_windows_core.txt index 24212bc0d32f..c7a19d5def7e 100644 --- a/tools/tdgpt/requirements_windows_core.txt +++ b/tools/tdgpt/requirements_windows_core.txt @@ -2,7 +2,7 @@ transformers==4.51.0 --find-links https://download.pytorch.org/whl/torch/ -torch==2.6.0+cpu +torch==2.12.1 accelerate==0.34.2 cookiecutter>=2.1.1 ray>=2.9.0