xinjie.wang commited on
Commit
13b3610
·
1 Parent(s): 2e90551
Files changed (1) hide show
  1. requirements.txt +5 -5
requirements.txt CHANGED
@@ -1,9 +1,9 @@
1
- # --extra-index-url https://download.pytorch.org/whl/cu118
2
 
3
- torch==2.1.0+cu118
4
- torchaudio==2.1.0+cu118
5
- torchvision==0.16.0+cu118
6
- xformers==0.0.22.post7+cu118
7
  dataclasses_json
8
  easydict
9
  opencv-python>4.5
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu118
2
 
3
+ torch==2.1.0
4
+ torchaudio==2.1.0
5
+ torchvision==0.16.0
6
+ xformers==0.0.22.post7
7
  dataclasses_json
8
  easydict
9
  opencv-python>4.5