ZhouZJ36DL commited on
Commit
58e3f0a
·
verified ·
1 Parent(s): 8db1226

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -1,4 +1,4 @@
1
- torch==2.6.0
2
  einops
3
  accelerate==1.4.0
4
  deepspeed==0.14.4
@@ -12,7 +12,7 @@ sentencepiece
12
  opencv-python
13
  matplotlib
14
  onnxruntime
15
- torchvision==0.21.0
16
  timm
17
  invisible-watermark
18
  fire
 
1
+ torch
2
  einops
3
  accelerate==1.4.0
4
  deepspeed==0.14.4
 
12
  opencv-python
13
  matplotlib
14
  onnxruntime
15
+ torchvision
16
  timm
17
  invisible-watermark
18
  fire