phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-30 03:29:31 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add datasetio code

Browse source
This commit is contained in:
raspawar 2025-03-26 15:04:20 +05:30 • committed by Rashmi Pawar
parent c149cf2e0f
commit ae973c9595
6 changed files with 220 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -36,4 +36,15 @@ def available_providers() -> List[ProviderSpec]:
config_class="llama_stack.providers.remote.datasetio.huggingface.HuggingfaceDatasetIOConfig",
),
),
remote_provider_spec(
api=Api.datasetio,
adapter=AdapterSpec(
adapter_type="nvidia",
pip_packages=[
"datasets",
],
module="llama_stack.providers.remote.datasetio.nvidia",
config_class="llama_stack.providers.remote.datasetio.nvidia.NvidiaDatasetIOConfig",
),
),
]