Peter Larnholt commited on
Commit
1b713bd
·
1 Parent(s): 65659d1

Upgrade torch to 2.5.1 for vLLM 0.6.4.post1 compatibility

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -6,6 +6,6 @@ requests>=2.31
6
  # vLLM + CUDA 12.1
7
  vllm==0.6.4.post1
8
  --extra-index-url https://download.pytorch.org/whl/cu121
9
- torch==2.4.0
10
  transformers>=4.44
11
  accelerate>=0.30
 
6
  # vLLM + CUDA 12.1
7
  vllm==0.6.4.post1
8
  --extra-index-url https://download.pytorch.org/whl/cu121
9
+ torch==2.5.1
10
  transformers>=4.44
11
  accelerate>=0.30