update readme model registration code

This commit is contained in:
Rashmi Pawar 2025-04-16 14:22:52 +05:30
parent 6c2153a6d0
commit 436b3269dc

View file

@ -128,13 +128,14 @@ client.post_training.job.cancel(job_uuid="your-job-id")
#### 1. Register the model #### 1. Register the model
```python ```python
model = Model( from llama_stack.apis.models import Model, ModelType
identifier="test-example-model@v1",
client.models.register(
model_id="test-example-model@v1",
provider_id="nvidia", provider_id="nvidia",
provider_model_id="test-example-model@v1", provider_model_id="test-example-model@v1",
model_type=ModelType.llm, model_type=ModelType.llm,
) )
client.register_model(model)
``` ```
#### 2. Inference with the fine-tuned model #### 2. Inference with the fine-tuned model