diff --git a/backend/python/autogptq/requirements-cublas11.txt b/backend/python/autogptq/requirements-cublas11.txt index cf4694727e7..c57164dda28 100644 --- a/backend/python/autogptq/requirements-cublas11.txt +++ b/backend/python/autogptq/requirements-cublas11.txt @@ -1,2 +1,2 @@ --extra-index-url https://download.pytorch.org/whl/cu118 -torch==2.4.1+cu118 +torch==2.5.1+cu118 diff --git a/backend/python/autogptq/requirements-cublas12.txt b/backend/python/autogptq/requirements-cublas12.txt index 20f84cf7bcf..f702c8a3f0c 100644 --- a/backend/python/autogptq/requirements-cublas12.txt +++ b/backend/python/autogptq/requirements-cublas12.txt @@ -1 +1 @@ -torch==2.4.1 \ No newline at end of file +torch==2.5.1+cu118 \ No newline at end of file diff --git a/backend/python/autogptq/requirements-hipblas.txt b/backend/python/autogptq/requirements-hipblas.txt index ecd817dcc2d..69ef6e99908 100644 --- a/backend/python/autogptq/requirements-hipblas.txt +++ b/backend/python/autogptq/requirements-hipblas.txt @@ -1,2 +1,2 @@ --extra-index-url https://download.pytorch.org/whl/rocm6.0 -torch==2.4.1+rocm6.0 \ No newline at end of file +torch==2.5.1+cu118 \ No newline at end of file