forked from phoenix-oss/llama-stack-mirror
add embedding model by default to distribution templates (#617)
# What does this PR do? Adds the sentence transformer provider and the `all-MiniLM-L6-v2` embedding model to the default models to register in the run.yaml for all providers. ## Test Plan llama stack build --template together --image-type conda llama stack run ~/.llama/distributions/llamastack-together/together-run.yaml
This commit is contained in:
parent
e893b22868
commit
516e1a3e59
41 changed files with 473 additions and 64 deletions
|
@ -21,9 +21,10 @@ class CommonModelFields(BaseModel):
|
|||
)
|
||||
|
||||
|
||||
class ModelType(Enum):
|
||||
@json_schema_type
|
||||
class ModelType(str, Enum):
|
||||
llm = "llm"
|
||||
embedding_model = "embedding"
|
||||
embedding = "embedding"
|
||||
|
||||
|
||||
@json_schema_type
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue