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

instantiate inference models

Browse source
This commit is contained in:
Xi Yan 2024-09-19 21:33:35 -07:00
parent d2ec822b12
commit 7071c46422
6 changed files with 40 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_stack/providers/routers/inference/__init__.py
View file

@ -9,9 +9,9 @@ from typing import Any, List, Tuple
from llama_stack.distribution.datatypes import Api
async def get_router_impl(inner_impls: List[Tuple[str, Any]], deps: List[Api]):
async def get_router_impl(models_api: Api):
from .inference import InferenceRouterImpl
impl = InferenceRouterImpl(inner_impls, deps)
impl = InferenceRouterImpl(models_api)
await impl.initialize()
return impl