QinLei086 commited on
Commit
d9496a8
1 Parent(s): 5751a82

change requirements

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -3
requirements.txt CHANGED
@@ -163,11 +163,11 @@ toml==0.10.2
163
  tomli==2.0.1
164
  tomlkit==0.12.0
165
  toolz==0.12.0
166
- torch==2.1.0+cu118
167
  torch-tb-profiler==0.4.3
168
- torchaudio==2.1.0+cu118
169
  torchmetrics==1.2.0
170
- torchvision==0.16.0+cu118
171
  tqdm==4.66.1
172
  transformers==4.26.1
173
  triton==2.1.0
 
163
  tomli==2.0.1
164
  tomlkit==0.12.0
165
  toolz==0.12.0
166
+ torch==2.1.0
167
  torch-tb-profiler==0.4.3
168
+ torchaudio==2.1.0
169
  torchmetrics==1.2.0
170
+ torchvision==0.16.0
171
  tqdm==4.66.1
172
  transformers==4.26.1
173
  triton==2.1.0