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

try upgrading torchao / torchtune deps

Browse source
This commit is contained in:
Ashwin Bharambe 2025-08-26 13:20:49 -07:00
parent 104c66f099
commit 967ac345e0
2 changed files with 6 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/providers/registry/inference.py
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 --index-url https://download.pytorch.org/whl/cpu",
"torch torchvision torchao>=0.12.0 --index-url https://download.pytorch.org/whl/cpu",
"sentence-transformers --no-deps",
],
module="llama_stack.providers.inline.inference.sentence_transformers",