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}