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

huggingface provider

Browse source
This commit is contained in:
Xi Yan 2024-11-07 10:21:25 -08:00
parent 283b5c1def
commit d75095033d
4 changed files with 117 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

8
llama_stack/providers/registry/datasetio.py
View file

@ -19,4 +19,12 @@ def available_providers() -> List[ProviderSpec]:
config_class="llama_stack.providers.inline.meta_reference.datasetio.MetaReferenceDatasetIOConfig",
api_dependencies=[],
),
InlineProviderSpec(
api=Api.datasetio,
provider_type="huggingface",
pip_packages=["datasets"],
module="llama_stack.providers.inline.huggingface.datasetio",
config_class="llama_stack.providers.inline.huggingface.datasetio.HuggingfaceDatasetIOConfig",
api_dependencies=[],
),
]