nahid112376 commited on
Commit
8303482
·
1 Parent(s): 3248dd8

Add torchvision dependency

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -0
requirements.txt CHANGED
@@ -2,6 +2,7 @@
2
 
3
  # Core ML frameworks
4
  torch>=2.0.0
 
5
  transformers>=4.37.0
6
  accelerate>=0.25.0
7
 
 
2
 
3
  # Core ML frameworks
4
  torch>=2.0.0
5
+ torchvision>=0.15.0
6
  transformers>=4.37.0
7
  accelerate>=0.25.0
8