From 436b3269dc549e34ba31e8f03dde025eac0398b1 Mon Sep 17 00:00:00 2001
From: Rashmi Pawar <raspawar@nvidia.com>
Date: Wed, 16 Apr 2025 14:22:52 +0530
Subject: [PATCH] update readme model registration code

---
 .../providers/remote/post_training/nvidia/README.md        | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/llama_stack/providers/remote/post_training/nvidia/README.md b/llama_stack/providers/remote/post_training/nvidia/README.md
index 015d6a5f9..13164667f 100644
--- a/llama_stack/providers/remote/post_training/nvidia/README.md
+++ b/llama_stack/providers/remote/post_training/nvidia/README.md
@@ -128,13 +128,14 @@ client.post_training.job.cancel(job_uuid="your-job-id")
 #### 1. Register the model
 
 ```python
-model = Model(
-    identifier="test-example-model@v1",
+from llama_stack.apis.models import Model, ModelType
+
+client.models.register(
+    model_id="test-example-model@v1",
     provider_id="nvidia",
     provider_model_id="test-example-model@v1",
     model_type=ModelType.llm,
 )
-client.register_model(model)
 ```
 
 #### 2. Inference with the fine-tuned model