Update install_pytorch.sh

GregoryComer · web-flow · commit 584dc6004f12 · 2025-10-15T12:50:15.000-07:00
diff --git a/.ci/docker/common/install_pytorch.sh b/.ci/docker/common/install_pytorch.sh
@@ -16,7 +16,7 @@ install_domains() {
 }
 
 install_pytorch_and_domains() {
-  pip_install torch==2.9.0 torchvision=0.24.0 torchaudio=2.9.0 torchao==0.14.0
+  pip_install torch==2.9.0 torchvision==0.24.0 torchaudio==2.9.0 torchao==0.14.0
 }
 
 install_pytorch_and_domains

Original file line number	Diff line number	Diff line change
`@@ -16,7 +16,7 @@ install_domains() {`
`16`	`16`	`}`
`17`	`17`
`18`	`18`	`install_pytorch_and_domains() {`
`19`		`- pip_install torch==2.9.0 torchvision=0.24.0 torchaudio=2.9.0 torchao==0.14.0`
	`19`	`+ pip_install torch==2.9.0 torchvision==0.24.0 torchaudio==2.9.0 torchao==0.14.0`
`20`	`20`	`}`
`21`	`21`
`22`	`22`	`install_pytorch_and_domains`