Files changed (1) hide show
  1. requirements.txt +2 -1
requirements.txt CHANGED
@@ -1,3 +1,4 @@
1
  torch
2
  torchvision
3
- lightning
 
 
1
  torch
2
  torchvision
3
+ lightning
4
+ onnxruntime