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

add register model to unit test

Browse source
This commit is contained in:
Xi Yan 2024-11-11 10:35:59 -05:00
parent e690eb7ad3
commit 1031f1404b
8 changed files with 23 additions and 89 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -23,8 +23,8 @@ def available_providers() -> List[ProviderSpec]:
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",
module="llama_stack.providers.adapters.datasetio.huggingface",
config_class="llama_stack.providers.adapters.datasetio.huggingface.HuggingfaceDatasetIOConfig",
api_dependencies=[],
),
]