build: added missing transformer dep

mbari-org · Dec 13, 2024 · 4300da1 · 4300da1
1 parent 7012c8f
commit 4300da1
Show file tree

Hide file tree

Showing 3 changed files with 111 additions and 2 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -24,6 +24,7 @@ scipy = "^1.14.1"
 open-clip-torch = "^2.29.0"
 grad-cam = "^1.5.4"
 numpy = "1.26.4"
+transformers = "^4.47.0"
 
 
 [build-system]

diff --git a/requirements.txt b/requirements.txt
@@ -54,9 +54,11 @@ six==1.16.0 ; python_version >= "3.10" and python_version < "3.12"
 sympy==1.13.1 ; python_version >= "3.10" and python_version < "3.12"
 threadpoolctl==3.5.0 ; python_version >= "3.10" and python_version < "3.12"
 timm==1.0.11 ; python_version >= "3.10" and python_version < "3.12"
+tokenizers==0.21.0 ; python_version >= "3.10" and python_version < "3.12"
 torch==2.5.1 ; python_version >= "3.10" and python_version < "3.12"
 torchvision==0.20.1 ; python_version >= "3.10" and python_version < "3.12"
 tqdm==4.67.0 ; python_version >= "3.10" and python_version < "3.12"
+transformers==4.47.0 ; python_version >= "3.10" and python_version < "3.12"
 triton==3.1.0 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version < "3.12" and python_version >= "3.10"
 ttach==0.0.3 ; python_version >= "3.10" and python_version < "3.12"
 typing-extensions==4.12.2 ; python_version >= "3.10" and python_version < "3.12"