diff --git a/llama_stack/providers/registry/post_training.py b/llama_stack/providers/registry/post_training.py index 5c93a2e22..67238e3fc 100644 --- a/llama_stack/providers/registry/post_training.py +++ b/llama_stack/providers/registry/post_training.py @@ -32,7 +32,7 @@ def available_providers() -> list[ProviderSpec]: "provider_type": "inline::torchtune-cpu", "pip_packages": ( cast(list[str], torchtune_def["pip_packages"]) - + ["torch torchtune>=0.5.0 torchao>=0.12.0 --index-url https://download.pytorch.org/whl/cpu"] + + ["torch torchtune>=0.5.0 torchao>=0.12.0 --extra-index-url https://download.pytorch.org/whl/cpu"] ), }, ),