fix safety using inference (#99)

This commit is contained in:
Xi Yan 2024-09-25 11:30:27 -07:00 committed by GitHub
parent 5c4f73d52f
commit 82f420c4f0
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 4 additions and 0 deletions

View file

@ -28,6 +28,9 @@ def available_providers() -> List[ProviderSpec]:
],
module="llama_stack.providers.impls.meta_reference.safety",
config_class="llama_stack.providers.impls.meta_reference.safety.SafetyConfig",
api_dependencies=[
Api.inference,
],
),
remote_provider_spec(
api=Api.safety,