phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-07-29 07:14:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix precommit

Browse source 
Signed-off-by: Yuan Tang <terrytangyuan@gmail.com>
This commit is contained in:
Yuan Tang 2024-10-10 21:08:19 -04:00
parent 59b4a4bbe5
commit d7c0d2c46f
No known key found for this signature in database
4 changed files with 10 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

2
docs/getting_started.md
View file

@ -73,7 +73,7 @@ docker run -it -p 5000:5000 -v ~/.llama:/root/.llama --gpus=all llamastack-local
``` ```
> [!NOTE] > [!NOTE]
> `~/.llama` should be the path containing downloaded weights of Llama models. > `~/.llama` should be the path containing downloaded weights of Llama models.
#### Via conda #### Via conda

3
llama_stack/providers/adapters/inference/databricks/__init__.py
View file

@ -7,10 +7,11 @@
from .config import DatabricksImplConfig from .config import DatabricksImplConfig
from .databricks import DatabricksInferenceAdapter from .databricks import DatabricksInferenceAdapter
async def get_adapter_impl(config: DatabricksImplConfig, _deps): async def get_adapter_impl(config: DatabricksImplConfig, _deps):
assert isinstance( assert isinstance(
config, DatabricksImplConfig config, DatabricksImplConfig
), f"Unexpected config type: {type(config)}" ), f"Unexpected config type: {type(config)}"
impl = DatabricksInferenceAdapter(config) impl = DatabricksInferenceAdapter(config)
await impl.initialize() await impl.initialize()
return impl return impl

2
llama_stack/providers/adapters/inference/databricks/config.py
View file

@ -19,4 +19,4 @@ class DatabricksImplConfig(BaseModel):
api_token: str = Field( api_token: str = Field(
default=None, default=None,
description="The Databricks API token", description="The Databricks API token",
) )

6
llama_stack/providers/impls/vllm/__init__.py
View file

@ -1,3 +1,9 @@
# Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.
#
# This source code is licensed under the terms described in the LICENSE file in
# the root directory of this source tree.
from typing import Any from typing import Any
from .config import VLLMConfig from .config import VLLMConfig