use extra-index instead of index

This commit is contained in:
Ashwin Bharambe 2025-08-26 13:23:17 -07:00
parent 967ac345e0
commit 360522969c

View file

@ -32,7 +32,7 @@ def available_providers() -> list[ProviderSpec]:
"provider_type": "inline::torchtune-cpu", "provider_type": "inline::torchtune-cpu",
"pip_packages": ( "pip_packages": (
cast(list[str], torchtune_def["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"]
), ),
}, },
), ),