maxiw commited on
Commit
ac09f39
1 Parent(s): a02f343

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -2,7 +2,7 @@ flash_attn==2.5.8
2
  numpy==1.24.4
3
  Pillow==10.3.0
4
  Requests==2.31.0
5
- torch==2.3.0
6
- torchvision==0.18.0
7
  transformers==4.43.0
8
  accelerate==0.30.0
 
2
  numpy==1.24.4
3
  Pillow==10.3.0
4
  Requests==2.31.0
5
+ torch
6
+ torchvision
7
  transformers==4.43.0
8
  accelerate==0.30.0