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
|
from .config import HuggingfaceDatasetIOConfig
|
||||||
|
|
||||||
|
|
||||||
async def get_provider_impl(
|
async def get_adapter_impl(
|
||||||
config: HuggingfaceDatasetIOConfig,
|
config: HuggingfaceDatasetIOConfig,
|
||||||
_deps,
|
_deps,
|
||||||
):
|
):
|
||||||
|
|
|
@ -19,12 +19,15 @@ def available_providers() -> List[ProviderSpec]:
|
||||||
config_class="llama_stack.providers.inline.meta_reference.datasetio.MetaReferenceDatasetIOConfig",
|
config_class="llama_stack.providers.inline.meta_reference.datasetio.MetaReferenceDatasetIOConfig",
|
||||||
api_dependencies=[],
|
api_dependencies=[],
|
||||||
),
|
),
|
||||||
InlineProviderSpec(
|
remote_provider_spec(
|
||||||
api=Api.datasetio,
|
api=Api.datasetio,
|
||||||
provider_type="huggingface",
|
adapter=AdapterSpec(
|
||||||
pip_packages=["datasets"],
|
adapter_type="huggingface",
|
||||||
module="llama_stack.providers.adapters.datasetio.huggingface",
|
pip_packages=[
|
||||||
config_class="llama_stack.providers.adapters.datasetio.huggingface.HuggingfaceDatasetIOConfig",
|
"datasets",
|
||||||
api_dependencies=[],
|
],
|
||||||
|
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=[
|
providers=[
|
||||||
Provider(
|
Provider(
|
||||||
provider_id="huggingface",
|
provider_id="huggingface",
|
||||||
provider_type="huggingface",
|
provider_type="remote::huggingface",
|
||||||
config={},
|
config={},
|
||||||
)
|
)
|
||||||
],
|
],
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue