wzhouxiff commited on
Commit
81c3c04
1 Parent(s): e25695f

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -11,8 +11,8 @@ scikit-learn
11
  taming-transformers-rom1504
12
  pandas
13
  triton
14
- xformers==0.0.16
15
- torch==1.13.1
16
  torchvision
17
  fairscale
18
  psutil==5.9.5
 
11
  taming-transformers-rom1504
12
  pandas
13
  triton
14
+ xformers==0.0.20
15
+ torch==2.0.1
16
  torchvision
17
  fairscale
18
  psutil==5.9.5