diff --git a/llama_stack/templates/nvidia/nvidia.py b/llama_stack/templates/nvidia/nvidia.py index 44cd7dae0..32ddf78e3 100644 --- a/llama_stack/templates/nvidia/nvidia.py +++ b/llama_stack/templates/nvidia/nvidia.py @@ -65,7 +65,7 @@ def get_distribution_template() -> DistributionTemplate: default_models = get_model_registry(available_models) return DistributionTemplate( name="nvidia", - distro_type="remote_hosted", + distro_type="self_hosted", description="Use NVIDIA NIM for running LLM inference, evaluation and safety", container_image=None, template_path=Path(__file__).parent / "doc_template.md",