Peter Larnholt commited on
Commit
972cd35
·
1 Parent(s): e48919a

Fix torch version to 2.4.0 for vLLM 0.6.3.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.3.post1
8
  --extra-index-url https://download.pytorch.org/whl/cu121
9
- torch==2.5.0
10
  transformers>=4.44
11
  accelerate>=0.30
 
6
  # vLLM + CUDA 12.1
7
  vllm==0.6.3.post1
8
  --extra-index-url https://download.pytorch.org/whl/cu121
9
+ torch==2.4.0
10
  transformers>=4.44
11
  accelerate>=0.30