fix template validation

This commit is contained in:
Rashmi Pawar 2025-04-28 16:04:57 +05:30
parent 00e57d693f
commit aa8b2aa31f
4 changed files with 6 additions and 9 deletions

View file

@ -6,7 +6,7 @@ The `llamastack/distribution-nvidia` distribution consists of the following prov
| API | Provider(s) | | API | Provider(s) |
|-----|-------------| |-----|-------------|
| agents | `inline::meta-reference` | | agents | `inline::meta-reference` |
| datasetio | `inline::localfs` | | datasetio | `inline::localfs`, `remote::nvidia` |
| eval | `remote::nvidia` | | eval | `remote::nvidia` |
| inference | `remote::nvidia` | | inference | `remote::nvidia` |
| post_training | `remote::nvidia` | | post_training | `remote::nvidia` |

View file

@ -395,7 +395,6 @@
"blobfile", "blobfile",
"chardet", "chardet",
"datasets", "datasets",
"emoji",
"faiss-cpu", "faiss-cpu",
"fastapi", "fastapi",
"fire", "fire",

View file

@ -39,6 +39,11 @@ def get_distribution_template() -> DistributionTemplate:
provider_type="remote::nvidia", provider_type="remote::nvidia",
config=NVIDIASafetyConfig.sample_run_config(), config=NVIDIASafetyConfig.sample_run_config(),
) )
datasetio_provider = Provider(
provider_id="nvidia",
provider_type="remote::nvidia",
config=NvidiaDatasetIOConfig.sample_run_config(),
)
eval_provider = Provider( eval_provider = Provider(
provider_id="nvidia", provider_id="nvidia",
provider_type="remote::nvidia", provider_type="remote::nvidia",

View file

@ -62,13 +62,6 @@ providers:
project_id: ${env.NVIDIA_PROJECT_ID:test-project} project_id: ${env.NVIDIA_PROJECT_ID:test-project}
customizer_url: ${env.NVIDIA_CUSTOMIZER_URL:http://nemo.test} customizer_url: ${env.NVIDIA_CUSTOMIZER_URL:http://nemo.test}
datasetio: datasetio:
- provider_id: localfs
provider_type: inline::localfs
config:
kvstore:
type: sqlite
namespace: null
db_path: ${env.SQLITE_STORE_DIR:~/.llama/distributions/nvidia}/localfs_datasetio.db
- provider_id: nvidia - provider_id: nvidia
provider_type: remote::nvidia provider_type: remote::nvidia
config: config: