rui3000 commited on
Commit
b34b79c
Β·
verified Β·
1 Parent(s): 6032be5

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -1
requirements.txt CHANGED
@@ -8,7 +8,10 @@ python-multipart>=0.0.9
8
  tabulate==0.9.0
9
  httpx
10
  transformers>=4.36.0
11
- torch==2.4.0
 
 
 
12
  huggingface-hub>=0.30.0,<1.0.0
13
  accelerate>=0.25.0
14
  sentencepiece>=0.1.99
 
8
  tabulate==0.9.0
9
  httpx
10
  transformers>=4.36.0
11
+ --extra-index-url https://download.pytorch.org/whl/cpu
12
+ torch==2.1.0+cpu
13
+ torchvision==0.16.0+cpu
14
+ torchaudio==2.1.0+cpu
15
  huggingface-hub>=0.30.0,<1.0.0
16
  accelerate>=0.25.0
17
  sentencepiece>=0.1.99