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

temp commit

Browse source
This commit is contained in:
Botao Chen 2024-11-26 21:23:56 -08:00
parent 90add9fed0
commit c31a78dfcb
7 changed files with 92 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_stack/providers/registry/post_training.py
View file

@ -12,6 +12,7 @@ from llama_stack.distribution.datatypes import * # noqa: F403
META_REFERENCE_DEPS = [
"torch",
"torchtune",
"torchao",
"numpy",
]
@ -24,5 +25,8 @@ def available_providers() -> List[ProviderSpec]:
pip_packages=META_REFERENCE_DEPS,
module="llama_stack.providers.inline.post_training.meta_reference",
config_class="llama_stack.providers.inline.post_training.meta_reference.MetaReferencePostTrainingConfig",
api_dependencies=[
Api.datasetio,
],
),
]