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

Add vLLM provider

Browse source 
Signed-off-by: Yuan Tang <terrytangyuan@gmail.com>
This commit is contained in:
Yuan Tang 2024-10-03 13:01:56 -04:00
parent 95a96afe34
commit 925e1afb5b
No known key found for this signature in database
4 changed files with 317 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -60,6 +60,14 @@ def available_providers() -> List[ProviderSpec]:
module="llama_stack.providers.adapters.inference.ollama",
),
),
remote_provider_spec(
api=Api.inference,
adapter=AdapterSpec(
adapter_type="vllm",
pip_packages=["openai"],
module="llama_stack.providers.adapters.inference.vllm",
),
),
remote_provider_spec(
api=Api.inference,
adapter=AdapterSpec(