decodingchris commited on
Commit
1cb2029
·
verified ·
1 Parent(s): f0e2fd2

Remove specific torch version, added accelerate

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -2
requirements.txt CHANGED
@@ -4,5 +4,6 @@ numpy==1.26.4
4
  pandas==2.2.2
5
  piexif==1.1.3
6
  pillow==11.0.0
7
- torch==2.5.1
8
- torchvision==0.20.1
 
 
4
  pandas==2.2.2
5
  piexif==1.1.3
6
  pillow==11.0.0
7
+ torch
8
+ torchvision==0.20.1
9
+ accelerate