update deps, mypy

This commit is contained in:
Ashwin Bharambe 2025-10-20 22:20:16 -07:00
parent 16ec064cc8
commit 1fa6fe6101
2 changed files with 3 additions and 2 deletions

View file

@ -42,6 +42,7 @@ def available_providers() -> list[ProviderSpec]:
# CrossEncoder depends on torchao.quantization
pip_packages=[
"torch torchvision torchao>=0.12.0 --extra-index-url https://download.pytorch.org/whl/cpu",
"numpy tqdm transformers",
"sentence-transformers --no-deps",
# required by some SentenceTransformers architectures for tensor rearrange/merge ops
"einops",