diff --git a/requirements.txt b/requirements.txt index 1a42a3b..b45846c 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,3 +1,7 @@ -torch towhee>=0.7.1 towhee.models>=0.7.1 +einops>=0.4.1 +transformers>=4.19.2 +timm>=0.4.12 +torchvision +torch \ No newline at end of file