phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2026-01-03 11:52:16 +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",
),
),
]

30
llama_stack/providers/registry/datasets.py
View file

@ -1,30 +0,0 @@
# Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.
#
# This source code is licensed under the terms described in the LICENSE file in
# the root directory of this source tree.
from typing import List
from llama_stack.providers.datatypes import (
AdapterSpec,
Api,
ProviderSpec,
remote_provider_spec,
)
def available_providers() -> List[ProviderSpec]:
return [
remote_provider_spec(
api=Api.datasets,
adapter=AdapterSpec(
adapter_type="nvidia",
pip_packages=[
"datasets",
],
module="llama_stack.providers.remote.datasets.nvidia",
config_class="llama_stack.providers.remote.datasets.nvidia.NvidiaDatasetConfig",
),
),
]