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

huggingface -> remote adapter

Browse source
This commit is contained in:
Xi Yan 2024-11-11 12:02:17 -05:00
parent e9a9ecb2dc
commit fdfc37a878
3 changed files with 11 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

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

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