add timm as optional

michaelfeil · Jun 13, 2024 · 32f622d · 32f622d
1 parent ad743bb
commit 32f622d
Show file tree

Hide file tree

Showing 2 changed files with 60 additions and 3 deletions.
diff --git a/libs/infinity_emb/poetry.lock b/libs/infinity_emb/poetry.lock
diff --git a/libs/infinity_emb/pyproject.toml b/libs/infinity_emb/pyproject.toml
@@ -29,7 +29,9 @@ ctranslate2 = {version = "^4.0.0", optional=true}
 optimum = {version = ">=1.16.2", optional=true, extras=["onnxruntime"]}
 hf_transfer = {version=">=0.1.5"}
 einops = {version = "*", optional=true}
+# vision
 pillow = {version = "*", optional=true}
+timm = {version = "*", optional=true}
 # cache
 diskcache = {version = "*", optional=true}
 # gpu
@@ -78,7 +80,7 @@ torch=["sentence-transformers","torch"]
 einops=["einops"]
 logging=["rich"]
 cache=["diskcache"]
-vision=["pillow"]
+vision=["pillow","timm"]
 server=[
     "fastapi",
     "orjson",