fffiloni commited on
Commit
d5a8d3d
1 Parent(s): d0b850a

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -4
requirements.txt CHANGED
@@ -1,4 +1,4 @@
1
- accelerate==0.20.3
2
  addict==2.4.0
3
  basicsr==1.4.2
4
  bitsandbytes==0.35.4
@@ -31,9 +31,9 @@ termcolor==2.2.0
31
  thinc==8.1.10
32
  timm==0.6.12
33
  tokenizers==0.13.2
34
- --extra-index-url https://download.pytorch.org/whl/cu117
35
- torch==1.13.1+cu117
36
- torchvision==0.14.1+cu117
37
  tqdm==4.64.1
38
  transformers==4.26.1
39
  triton
 
1
+ accelerate==0.17.1
2
  addict==2.4.0
3
  basicsr==1.4.2
4
  bitsandbytes==0.35.4
 
31
  thinc==8.1.10
32
  timm==0.6.12
33
  tokenizers==0.13.2
34
+ --extra-index-url https://download.pytorch.org/whl/cu116
35
+ torch==1.13.1+cu116
36
+ torchvision==0.14.1+cu116
37
  tqdm==4.64.1
38
  transformers==4.26.1
39
  triton