baubin commited on
Commit
b126191
·
1 Parent(s): c812906
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -9,7 +9,7 @@ peft==0.6.0
9
  pydantic==2.11.7
10
  spaces
11
  transformers >= 4.34.0
12
- --extra-index-url https://download.pytorch.org/whl/cu118
13
  torch==2.1.0
14
  torchvision>=0.16.0
15
  torchaudio>=2.1.0
 
9
  pydantic==2.11.7
10
  spaces
11
  transformers >= 4.34.0
12
+ --extra-index-url https://download.pytorch.org/whl/cu121
13
  torch==2.1.0
14
  torchvision>=0.16.0
15
  torchaudio>=2.1.0