phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-10 04:08:31 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

rm short context emb models

Browse source
This commit is contained in:
Xi Yan 2025-02-15 14:52:07 -08:00
parent 2e96a9299a
commit 532f607941
3 changed files with 0 additions and 76 deletions
Download patch file Download diff file Expand all files Collapse all files

24
llama_stack/templates/together/run-with-safety.yaml
View file

@ -150,36 +150,12 @@ models:
provider_id: together
provider_model_id: togethercomputer/m2-bert-80M-8k-retrieval
model_type: embedding
- metadata:
embedding_dimension: 768
model_id: togethercomputer/m2-bert-80M-2k-retrieval
provider_id: together
provider_model_id: togethercomputer/m2-bert-80M-2k-retrieval
model_type: embedding
- metadata:
embedding_dimension: 768
model_id: togethercomputer/m2-bert-80M-32k-retrieval
provider_id: together
provider_model_id: togethercomputer/m2-bert-80M-32k-retrieval
model_type: embedding
- metadata:
embedding_dimension: 1024
model_id: WhereIsAI/UAE-Large-V1
provider_id: together
provider_model_id: WhereIsAI/UAE-Large-V1
model_type: embedding
- metadata:
embedding_dimension: 1024
model_id: BAAI/bge-large-en-v1.5
provider_id: together
provider_model_id: BAAI/bge-large-en-v1.5
model_type: embedding
- metadata:
embedding_dimension: 768
model_id: BAAI/bge-base-en-v1.5
provider_id: together
provider_model_id: BAAI/bge-base-en-v1.5
model_type: embedding
shields:
- shield_id: meta-llama/Llama-Guard-3-8B
provider_id: llama-guard

24
llama_stack/templates/together/run.yaml
View file

@ -144,36 +144,12 @@ models:
provider_id: together
provider_model_id: togethercomputer/m2-bert-80M-8k-retrieval
model_type: embedding
- metadata:
embedding_dimension: 768
model_id: togethercomputer/m2-bert-80M-2k-retrieval
provider_id: together
provider_model_id: togethercomputer/m2-bert-80M-2k-retrieval
model_type: embedding
- metadata:
embedding_dimension: 768
model_id: togethercomputer/m2-bert-80M-32k-retrieval
provider_id: together
provider_model_id: togethercomputer/m2-bert-80M-32k-retrieval
model_type: embedding
- metadata:
embedding_dimension: 1024
model_id: WhereIsAI/UAE-Large-V1
provider_id: together
provider_model_id: WhereIsAI/UAE-Large-V1
model_type: embedding
- metadata:
embedding_dimension: 1024
model_id: BAAI/bge-large-en-v1.5
provider_id: together
provider_model_id: BAAI/bge-large-en-v1.5
model_type: embedding
- metadata:
embedding_dimension: 768
model_id: BAAI/bge-base-en-v1.5
provider_id: together
provider_model_id: BAAI/bge-base-en-v1.5
model_type: embedding
shields:
- shield_id: meta-llama/Llama-Guard-3-8B
vector_dbs: []

28
llama_stack/templates/together/together.py
View file

@ -89,13 +89,6 @@ def get_distribution_template() -> DistributionTemplate:
provider_model_id="togethercomputer/m2-bert-80M-8k-retrieval",
metadata={"embedding_dimension": 768},
),
ModelInput(
model_id="togethercomputer/m2-bert-80M-2k-retrieval",
provider_id="together",
model_type=ModelType.embedding,
provider_model_id="togethercomputer/m2-bert-80M-2k-retrieval",
metadata={"embedding_dimension": 768},
),
ModelInput(
model_id="togethercomputer/m2-bert-80M-32k-retrieval",
provider_id="together",
@ -103,27 +96,6 @@ def get_distribution_template() -> DistributionTemplate:
provider_model_id="togethercomputer/m2-bert-80M-32k-retrieval",
metadata={"embedding_dimension": 768},
),
ModelInput(
model_id="WhereIsAI/UAE-Large-V1",
provider_id="together",
model_type=ModelType.embedding,
provider_model_id="WhereIsAI/UAE-Large-V1",
metadata={"embedding_dimension": 1024},
),
ModelInput(
model_id="BAAI/bge-large-en-v1.5",
provider_id="together",
model_type=ModelType.embedding,
provider_model_id="BAAI/bge-large-en-v1.5",
metadata={"embedding_dimension": 1024},
),
ModelInput(
model_id="BAAI/bge-base-en-v1.5",
provider_id="together",
model_type=ModelType.embedding,
provider_model_id="BAAI/bge-base-en-v1.5",
metadata={"embedding_dimension": 768},
),
]
return DistributionTemplate(