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

move nvidia to remote datasetio

Browse source
This commit is contained in:
raspawar 2025-03-26 13:46:37 +05:30
parent 0a2af0e2f8
commit 63825eb493
10 changed files with 47 additions and 96 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",
),
),
]