mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-07-31 16:01:46 +00:00
huggingface -> remote adapter
This commit is contained in:
parent
e9a9ecb2dc
commit
fdfc37a878
3 changed files with 11 additions and 8 deletions
|
@ -7,7 +7,7 @@
|
|||
from .config import HuggingfaceDatasetIOConfig
|
||||
|
||||
|
||||
async def get_provider_impl(
|
||||
async def get_adapter_impl(
|
||||
config: HuggingfaceDatasetIOConfig,
|
||||
_deps,
|
||||
):
|
||||
|
|
|
@ -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",
|
||||
),
|
||||
),
|
||||
]
|
||||
|
|
|
@ -37,7 +37,7 @@ def datasetio_huggingface() -> ProviderFixture:
|
|||
providers=[
|
||||
Provider(
|
||||
provider_id="huggingface",
|
||||
provider_type="huggingface",
|
||||
provider_type="remote::huggingface",
|
||||
config={},
|
||||
)
|
||||
],
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue