Dracomoth commited on
Commit
1813d3c
·
1 Parent(s): b8e76cb

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +8 -2
requirements.txt CHANGED
@@ -1,2 +1,8 @@
1
- torch==1.13.1+cu117
2
- torchvision==0.14.1+cu117
 
 
 
 
 
 
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu118
2
+ torch==2.0.1+cu118
3
+ torchvision==0.15.2+cu118
4
+ torchaudio==2.0.2+cu118
5
+ torchtext==0.15.2
6
+ torchdata==0.6.1
7
+ xformers==0.0.20
8
+ triton==2.0.0