Remove the torchao integration (--torchao-config) (#34304)
Co-authored-by: Brayden Zhong <brayden@radixark.ai>
This commit is contained in:
co-authored by
Brayden Zhong
parent
fe0c18effd
commit
5e65dd01a7
@@ -413,7 +413,7 @@ uninstall_stale_flashinfer() {
|
||||
|
||||
install_pytorch_stack() {
|
||||
PYTORCH_SPECS=()
|
||||
for package in torch torchaudio torchvision torchao torchcodec; do
|
||||
for package in torch torchaudio torchvision torchcodec; do
|
||||
spec=$(grep -Po -m1 "\"${package}([<>=!~ ;][^\"]*)?\"" python/pyproject.toml | tr -d '"' || true)
|
||||
if [ -n "$spec" ]; then
|
||||
PYTORCH_SPECS+=("$spec")
|
||||
|
||||
Reference in New Issue
Block a user