This commit is contained in:
Sébastien Han 2025-06-20 10:17:23 +02:00
parent b6ebbe1bc0
commit 2a8b3e05f1
No known key found for this signature in database
3 changed files with 235 additions and 169 deletions

View file

@ -1,5 +1,5 @@
[build-system]
requires = ["setuptools>=80.0"]
requires = ["setuptools>=80.0", "numpy"]
build-backend = "setuptools.build_meta"
[project]
@ -26,7 +26,7 @@ dependencies = [
"fire",
"httpx",
"huggingface-hub",
"jinja2>=3.1.6",
"jinja2<=3.1.4",
"jsonschema",
"llama-stack-client>=0.2.11",
"openai>=1.66",
@ -911,7 +911,10 @@ watsonx = [
"uvicorn",
]
cpu = ["torch"]
gpu = ["torch"]
meta-reference-gpu-distro = ["llama_stack[meta-reference-gpu,gpu]"]
[dependency-groups]
dev = [
@ -944,7 +947,7 @@ unit = [
"sqlalchemy",
"sqlalchemy[asyncio]>=2.0.41",
"blobfile",
"faiss-cpu"
"faiss-cpu",
]
# These are the core dependencies required for running integration tests. They are shared across all
# providers. If a provider requires additional dependencies, please add them to your environment
@ -983,7 +986,7 @@ docs = [
"linkify",
"sphinxcontrib.openapi",
]
codegen = ["rich", "pydantic", "jinja2>=3.1.6", "tomlkit"]
codegen = ["rich", "pydantic", "tomlkit"]
[project.urls]
Homepage = "https://github.com/meta-llama/llama-stack"
@ -996,14 +999,32 @@ install-wheel-from-presigned = "llama_stack.cli.scripts.run:install_wheel_from_p
where = ["."]
include = ["llama_stack", "llama_stack.*"]
[[tool.uv.index]]
name = "pytorch-cpu"
url = "https://download.pytorch.org/whl/cpu"
explicit = true
# [[tool.uv.index]]
# name = "pytorch-cpu"
# url = "https://download.pytorch.org/whl/cpu"
# explicit = true
# [tool.uv.sources]
# torch = [{ index = "pytorch-cpu" }]
# torchvision = [{ index = "pytorch-cpu" }]
[tool.uv]
# Needed when building on macOS
conflicts = [[{ extra = "cpu" }, { extra = "gpu" }]]
[tool.uv.sources]
torch = [{ index = "pytorch-cpu" }]
torchvision = [{ index = "pytorch-cpu" }]
torch = [
{ index = "torch-cpu", extra = "cpu" },
{ index = "torch-gpu", extra = "gpu" },
]
[[tool.uv.index]]
name = "torch-cpu"
url = "https://download.pytorch.org/whl/cpu"
[[tool.uv.index]]
name = "torch-gpu"
url = "https://download.pytorch.org/whl/cu124"
[[tool.uv.dependency-metadata]]
name = "sentence-transformers"