From 0c4508d6c2755ce1df467c81327676c720821c6f Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Mon, 30 Dec 2024 15:52:12 -0800 Subject: [PATCH] revert safety --- .../templates/meta-reference-gpu/run-with-safety.yaml | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/llama_stack/templates/meta-reference-gpu/run-with-safety.yaml b/llama_stack/templates/meta-reference-gpu/run-with-safety.yaml index d2aafc4a3..deb6c4a91 100644 --- a/llama_stack/templates/meta-reference-gpu/run-with-safety.yaml +++ b/llama_stack/templates/meta-reference-gpu/run-with-safety.yaml @@ -13,7 +13,7 @@ apis: - telemetry providers: inference: - - provider_id: meta-reference + - provider_id: meta-reference-inference provider_type: inline::meta-reference config: model: ${env.INFERENCE_MODEL} @@ -22,7 +22,7 @@ providers: - provider_id: sentence-transformers provider_type: inline::sentence-transformers config: {} - - provider_id: meta-reference + - provider_id: meta-reference-safety provider_type: inline::meta-reference config: model: ${env.SAFETY_MODEL} @@ -84,12 +84,12 @@ metadata_store: models: - metadata: {} model_id: ${env.INFERENCE_MODEL} - provider_id: meta-reference + provider_id: meta-reference-inference provider_model_id: null model_type: llm - metadata: {} model_id: ${env.SAFETY_MODEL} - provider_id: meta-reference + provider_id: meta-reference-safety provider_model_id: null model_type: llm - metadata: