add requirement

3 years ago · 37098429e6
1 changed files with 6 additions and 1 deletions
--- a/requirements.txt
+++ b/requirements.txt
@ -1,3 +1,8 @@
 transformers>=4.19.2
 einops>=0.4.1
-timm>=0.4.12
+timm>=0.4.12
+torch
+towhee.models
+torch
+torchvision
+towhee