From 358db6ebc39889da115f9003bf31f3b4a2d305f7 Mon Sep 17 00:00:00 2001 From: Chantal D Gama Rose Date: Wed, 19 Feb 2025 16:36:56 -0800 Subject: [PATCH] adding nvidia to registry --- llama_stack/providers/registry/safety.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/llama_stack/providers/registry/safety.py b/llama_stack/providers/registry/safety.py index b9f7b6d78..3966b7d1d 100644 --- a/llama_stack/providers/registry/safety.py +++ b/llama_stack/providers/registry/safety.py @@ -85,4 +85,13 @@ Provider `inline::meta-reference` for API `safety` does not work with the latest config_class="llama_stack.providers.remote.safety.bedrock.BedrockSafetyConfig", ), ), + remote_provider_spec( + api=Api.safety, + adapter=AdapterSpec( + adapter_type="nvidia", + pip_packages=[], + module="llama_stack.providers.remote.safety.nvidia", + config_class="llama_stack.providers.remote.safety.nvida.NVIDIASafetyConfig", + ), + ), ]