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

local imports for faster cli

Browse source
This commit is contained in:
Hardik Shah 2024-08-02 16:34:29 -07:00
parent af4710c959
commit 67229f23a4
9 changed files with 44 additions and 47 deletions
Download patch file Download diff file Expand all files Collapse all files

5
llama_toolchain/distribution/registry.py
View file

@ -4,6 +4,7 @@
# This source code is licensed under the terms described in the LICENSE file in
# the root directory of this source tree.
from functools import lru_cache
from typing import List, Optional
from llama_toolchain.inference.adapters import available_inference_adapters
@ -27,7 +28,7 @@ COMMON_DEPENDENCIES = [
"hydra-core",
"hydra-zen",
"json-strong-typing",
"llama-models",
"git+ssh://git@github.com/meta-llama/llama-models.git",
"omegaconf",
"pandas",
"Pillow",
@ -43,6 +44,7 @@ COMMON_DEPENDENCIES = [
]
@lru_cache()
def available_distributions() -> List[Distribution]:
inference_adapters_by_id = {a.adapter_id: a for a in available_inference_adapters()}
@ -66,6 +68,7 @@ def available_distributions() -> List[Distribution]:
]
@lru_cache()
def resolve_distribution(name: str) -> Optional[Distribution]:
for dist in available_distributions():
if dist.name == name: