phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-17 13:12:36 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

check for extra-index-url to determine special deps

Browse source
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
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/core/build.py
View file

@ -80,7 +80,7 @@ def get_provider_dependencies(
normal_deps = [] normal_deps = []
special_deps = [] special_deps = []
for package in deps: for package in deps:
if "--no-deps" in package or "--index-url" in package: if any(f in package for f in ["--no-deps", "--index-url", "--extra-index-url"]):
special_deps.append(package) special_deps.append(package)
else: else:
normal_deps.append(package) normal_deps.append(package)

2
llama_stack/providers/registry/inference.py
View file

@ -42,7 +42,7 @@ def available_providers() -> list[ProviderSpec]:
provider_type="inline::sentence-transformers", provider_type="inline::sentence-transformers",
# CrossEncoder depends on torchao.quantization # CrossEncoder depends on torchao.quantization
pip_packages=[ 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", "sentence-transformers --no-deps",
], ],
module="llama_stack.providers.inline.inference.sentence_transformers", module="llama_stack.providers.inline.inference.sentence_transformers",