mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-17 19:09:48 +00:00
check for extra-index-url to determine special deps
This commit is contained in:
parent
360522969c
commit
bdf932ebd3
2 changed files with 2 additions and 2 deletions
|
|
@ -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",
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue