mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-08-03 01:03:59 +00:00
update readme model registration code
This commit is contained in:
parent
6c2153a6d0
commit
436b3269dc
1 changed files with 4 additions and 3 deletions
|
@ -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
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue