mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-08-05 10:13:05 +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
|
#### 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
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue