fix provider ids for together and fireworks

This commit is contained in:
Dinesh Yeduguru 2024-12-12 15:43:30 -08:00
parent 2f88006bd0
commit fe2eb39da7
27 changed files with 49 additions and 46 deletions

View file

@ -56,7 +56,7 @@ def get_distribution_template() -> DistributionTemplate:
provider_id="sentence-transformers",
model_type=ModelType.embedding,
metadata={
"embedding_dim": 384,
"embedding_dimension": 384,
},
)
safety_model = ModelInput(