diff --git a/llama_stack/providers/registry/eval.py b/llama_stack/providers/registry/eval.py index 51f93bd1b..9604d5da4 100644 --- a/llama_stack/providers/registry/eval.py +++ b/llama_stack/providers/registry/eval.py @@ -33,7 +33,7 @@ def available_providers() -> List[ProviderSpec]: "requests", ], module="llama_stack.providers.remote.eval.nvidia", - config_class="llama_stack.providers.remote.eval.nvidia.NVIDIAConfig", + config_class="llama_stack.providers.remote.eval.nvidia.NVIDIAEvalConfig", ), api_dependencies=[ Api.datasetio,