check for extra-index-url to determine special deps

This commit is contained in:
Ashwin Bharambe 2025-08-26 13:31:33 -07:00
parent 360522969c
commit bdf932ebd3
2 changed files with 2 additions and 2 deletions

View file

@ -42,7 +42,7 @@ def available_providers() -> list[ProviderSpec]:
provider_type="inline::sentence-transformers",
# CrossEncoder depends on torchao.quantization
pip_packages=[
"torch torchvision torchao>=0.12.0 --index-url https://download.pytorch.org/whl/cpu",
"torch torchvision torchao>=0.12.0 --extra-index-url https://download.pytorch.org/whl/cpu",
"sentence-transformers --no-deps",
],
module="llama_stack.providers.inline.inference.sentence_transformers",