From d667a7109f258101b8aea85c91fec8262f5a604d Mon Sep 17 00:00:00 2001
From: Ubuntu
 <ubuntu@llama-stack-customizer-dev-inst-2tx95fyisatvlic4we8hidx5tfj.us-central1-a.c.brevdevprod.internal>
Date: Wed, 12 Mar 2025 14:38:07 +0000
Subject: [PATCH] update docstring

---
 .../post_training/nvidia/post_training.py     | 28 +++++++++++++++++++
 1 file changed, 28 insertions(+)

diff --git a/llama_stack/providers/remote/post_training/nvidia/post_training.py b/llama_stack/providers/remote/post_training/nvidia/post_training.py
index 1c517c774..523211cac 100644
--- a/llama_stack/providers/remote/post_training/nvidia/post_training.py
+++ b/llama_stack/providers/remote/post_training/nvidia/post_training.py
@@ -203,6 +203,34 @@ class NvidiaPostTrainingAdapter(ModelRegistryHelper):
             - NVIDIA_ACCESS_POLICIES: Access policies for the project
             - NVIDIA_DATASET_NAMESPACE: Namespace of the dataset
             - NVIDIA_OUTPUT_MODEL_DIR: Directory to save the output model
+
+        Supported models:
+            - meta/llama-3.1-8b-instruct
+
+        Supported algorithm configs:
+            - LoRA, SFT
+
+        Supported Parameters:
+            - TrainingConfig:
+                - n_epochs
+                - data_config
+                - optimizer_config
+                - dtype
+                - efficiency_config
+                - max_steps_per_epoch
+            - DataConfig:
+                - dataset_id
+                - batch_size
+            - OptimizerConfig:
+                - lr
+            - LoRA config:
+                - adapter_dim
+                - adapter_dropout
+            Note:
+                - checkpoint_dir, hyperparam_search_config, logger_config are not supported atm, will be ignored
+                - output_model_dir is set via environment variable NVIDIA_OUTPUT_MODEL_DIR
+
+            User is informed about unsupported parameters via warnings.
         """
         # map model to nvidia model name
         nvidia_model = self.get_provider_model_id(model)