[RELEASE ONLY] Use torchao 0.14.0

GregoryComer · GregoryComer · commit cc248a1bb552 · 2025-10-13T14:29:08.000-07:00
diff --git a/.ci/docker/common/install_pytorch.sh b/.ci/docker/common/install_pytorch.sh
@@ -17,7 +17,7 @@ install_domains() {
 }
 
 install_pytorch_and_domains() {
-  pip_install torch==2.9.0 torchvision torchaudio --index-url https://download.pytorch.org/whl/test/cpu
+  pip_install torch==2.9.0 torchvision torchaudio torchao=0.14.0 --index-url https://download.pytorch.org/whl/test/cpu
 }
 
 install_pytorch_and_domains
diff --git a/install_requirements.py b/install_requirements.py
@@ -111,7 +111,6 @@ def install_requirements(use_pytorch_nightly):
     )
 
     LOCAL_REQUIREMENTS = [
-        "third-party/ao",  # We need the latest kernels for fast iteration, so not relying on pypi.
     ] + (
         [
             "extension/llm/tokenizers",  # TODO(larryliu0820): Setup a pypi package for this.
diff --git a/pyproject.toml b/pyproject.toml
@@ -78,6 +78,7 @@ dependencies=[
   "scikit-learn==1.7.1",
   "hydra-core>=1.3.0",
   "omegaconf>=2.3.0",
+  "torchao=0.14.0",
 ]
 
 [project.urls]

Original file line number	Diff line number	Diff line change
`@@ -17,7 +17,7 @@ install_domains() {`
`17`	`17`	`}`
`18`	`18`
`19`	`19`	`install_pytorch_and_domains() {`
`20`		`- pip_install torch==2.9.0 torchvision torchaudio --index-url https://download.pytorch.org/whl/test/cpu`
	`20`	`+ pip_install torch==2.9.0 torchvision torchaudio torchao=0.14.0 --index-url https://download.pytorch.org/whl/test/cpu`
`21`	`21`	`}`
`22`	`22`
`23`	`23`	`install_pytorch_and_domains`
Original file line number	Diff line number	Diff line change
`@@ -111,7 +111,6 @@ def install_requirements(use_pytorch_nightly):`
`111`	`111`	`)`
`112`	`112`
`113`	`113`	`LOCAL_REQUIREMENTS = [`
`114`		`- "third-party/ao", # We need the latest kernels for fast iteration, so not relying on pypi.`
`115`	`114`	`] + (`
`116`	`115`	`[`
`117`	`116`	`"extension/llm/tokenizers", # TODO(larryliu0820): Setup a pypi package for this.`