diff --git a/requirements-exllama.txt b/requirements-exllama.txt index dfbba80..b0a776e 100644 --- a/requirements-exllama.txt +++ b/requirements-exllama.txt @@ -2,4 +2,4 @@ safetensors==0.3.1 sentencepiece>=0.1.97 ninja==1.11.1 --find-links https://download.pytorch.org/whl/torch_stable.html -torch==2.0.1+cu118 \ No newline at end of file +torch==2.2.0 \ No newline at end of file diff --git a/requirements-transformer-embedding-cpu.txt b/requirements-transformer-embedding-cpu.txt index 821d857..ad6650b 100644 --- a/requirements-transformer-embedding-cpu.txt +++ b/requirements-transformer-embedding-cpu.txt @@ -1,2 +1,2 @@ --find-links https://download.pytorch.org/whl/torch_stable.html -torch==2.0.1 +torch==2.2.0 diff --git a/requirements-transformer-embedding-gpu.txt b/requirements-transformer-embedding-gpu.txt index 4b7264e..ad6650b 100644 --- a/requirements-transformer-embedding-gpu.txt +++ b/requirements-transformer-embedding-gpu.txt @@ -1,2 +1,2 @@ --find-links https://download.pytorch.org/whl/torch_stable.html -torch==2.0.1+cu118 +torch==2.2.0