mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-07-07 14:26:44 +00:00
25268854bc
# What does this PR do?
We were not using conditionals correctly, conditionals can only be used
when the env variable is set, so `${env.ENVIRONMENT:+}` would return
None is ENVIRONMENT is not set.
If you want to create a conditional value, you need to do
`${env.ENVIRONMENT:=}`, this will pick the value of ENVIRONMENT if set,
otherwise will return None.
Closes: https://github.com/meta-llama/llama-stack/issues/2564
Signed-off-by: Sébastien Han <seb@redhat.com>
1.1 KiB
1.1 KiB
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}