diff --git a/backend/python/exllama2/requirements-cpu.txt b/backend/python/exllama2/requirements-cpu.txt index 8bcc7065..1a20d18a 100644 --- a/backend/python/exllama2/requirements-cpu.txt +++ b/backend/python/exllama2/requirements-cpu.txt @@ -1,4 +1,4 @@ # https://github.com/huggingface/transformers/issues/38269#issuecomment-2900214777 transformers==4.52.1 accelerate -torch==2.4.1 \ No newline at end of file +torch==2.7.0+cu118 \ No newline at end of file diff --git a/backend/python/exllama2/requirements-cublas11.txt b/backend/python/exllama2/requirements-cublas11.txt index 4ea5fc1f..aa3b860d 100644 --- a/backend/python/exllama2/requirements-cublas11.txt +++ b/backend/python/exllama2/requirements-cublas11.txt @@ -1,5 +1,5 @@ --extra-index-url https://download.pytorch.org/whl/cu118 -torch==2.4.1+cu118 +torch==2.7.0+cu118 # https://github.com/huggingface/transformers/issues/38269#issuecomment-2900214777 transformers==4.52.1 accelerate \ No newline at end of file diff --git a/backend/python/exllama2/requirements-cublas12.txt b/backend/python/exllama2/requirements-cublas12.txt index 10e4abdc..2a678aef 100644 --- a/backend/python/exllama2/requirements-cublas12.txt +++ b/backend/python/exllama2/requirements-cublas12.txt @@ -1,4 +1,4 @@ -torch==2.4.1 +torch==2.7.0+cu118 # https://github.com/huggingface/transformers/issues/38269#issuecomment-2900214777 transformers==4.52.1 accelerate \ No newline at end of file