mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-07-27 06:28:50 +00:00
1.1 KiB
1.1 KiB
orphan |
---|
true |
remote::nvidia
Description
NVIDIA's post-training provider for fine-tuning models on NVIDIA's platform.
Configuration
Field | Type | Required | Default | Description |
---|---|---|---|---|
api_key |
str | None |
No | The NVIDIA API key. | |
dataset_namespace |
str | None |
No | default | The NVIDIA dataset namespace. |
project_id |
str | None |
No | test-example-model@v1 | The NVIDIA project ID. |
customizer_url |
str | None |
No | Base URL for the NeMo Customizer API | |
timeout |
<class 'int'> |
No | 300 | Timeout for the NVIDIA Post Training API |
max_retries |
<class 'int'> |
No | 3 | Maximum number of retries for the NVIDIA Post Training API |
output_model_dir |
<class 'str'> |
No | test-example-model@v1 | Directory to save the output model |
Sample Configuration
api_key: ${env.NVIDIA_API_KEY:=}
dataset_namespace: ${env.NVIDIA_DATASET_NAMESPACE:=default}
project_id: ${env.NVIDIA_PROJECT_ID:=test-project}
customizer_url: ${env.NVIDIA_CUSTOMIZER_URL:=http://nemo.test}