Make each inference provider into its own subdirectory

This commit is contained in:
Ashwin Bharambe 2024-08-05 15:13:52 -07:00
parent f64668319c
commit 0de5a807c7
42 changed files with 123 additions and 103 deletions

View file

@ -19,7 +19,7 @@ def available_safety_providers() -> List[ProviderSpec]:
"torch",
"transformers",
],
module="llama_toolchain.safety.safety",
config_class="llama_toolchain.safety.config.SafetyConfig",
module="llama_toolchain.safety.meta_reference",
config_class="llama_toolchain.safety.meta_reference.SafetyConfig",
),
]