diff --git a/tools/tdgpt/requirements_ess.txt b/tools/tdgpt/requirements_ess.txt index 742d7a5559b0..ac038b1d2994 100644 --- a/tools/tdgpt/requirements_ess.txt +++ b/tools/tdgpt/requirements_ess.txt @@ -1,6 +1,6 @@ -r requirements_docker.txt -transformers==4.40.0 +transformers==5.0.0rc3 --find-links https://download.pytorch.org/whl/torch/ torch==2.3.1+cpu accelerate==0.34.2 \ No newline at end of file diff --git a/tools/tdgpt/requirements_moment.txt b/tools/tdgpt/requirements_moment.txt index cfeb7790567f..e4aae5359758 100644 --- a/tools/tdgpt/requirements_moment.txt +++ b/tools/tdgpt/requirements_moment.txt @@ -1,6 +1,6 @@ -f https://download.pytorch.org/whl/torch_stable.html torch==2.3.1+cpu -transformers==4.33.3 +transformers==5.0.0rc3 numpy==1.25.2 matplotlib pandas==1.5 diff --git a/tools/tdgpt/requirements_windows_core.txt b/tools/tdgpt/requirements_windows_core.txt index 742d7a5559b0..ac038b1d2994 100644 --- a/tools/tdgpt/requirements_windows_core.txt +++ b/tools/tdgpt/requirements_windows_core.txt @@ -1,6 +1,6 @@ -r requirements_docker.txt -transformers==4.40.0 +transformers==5.0.0rc3 --find-links https://download.pytorch.org/whl/torch/ torch==2.3.1+cpu accelerate==0.34.2 \ No newline at end of file