llama-stack-mirror/llama_toolchain/distribution/registry.py
2024-08-01 14:44:57 -07:00

24 lines
771 B
Python

# 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 List
from .datatypes import LlamaStackDistribution
def all_registered_distributions() -> List[LlamaStackDistribution]:
return [
LlamaStackDistribution(
name="local-source",
description="Use code within `llama_toolchain` itself to run model inference and everything on top",
pip_packages=[],
),
LlamaStackDistribution(
name="local-ollama",
description="Like local-source, but use ollama for running LLM inference",
pip_packages=[],
),
]