diff --git a/pyproject.toml b/pyproject.toml
index 303d8ca..83a9801 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -39,6 +39,8 @@ dependencies = [
     "sentence-transformers",
     "torch",
     "numpy",
+    "timm",
+    "einops",
 ]
 
 [project.urls]