Yanlin Zhang commited on
Commit
a5e9f68
·
1 Parent(s): 899327c

add pytorch-vision in requirement

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -0
requirements.txt CHANGED
@@ -1,4 +1,5 @@
1
  torch
 
2
  transformers>=4.46.0
3
  accelerate
4
  gradio
 
1
  torch
2
+ torchvision
3
  transformers>=4.46.0
4
  accelerate
5
  gradio