File tree Expand file tree Collapse file tree 2 files changed +13
-11
lines changed
Expand file tree Collapse file tree 2 files changed +13
-11
lines changed Original file line number Diff line number Diff line change @@ -130,32 +130,34 @@ def install_requirements(use_pytorch_nightly):
130130
131131
132132def install_optional_example_requirements (use_pytorch_nightly ):
133- print ("Installing packages in requirements-examples.txt" )
133+ print ("Installing torch domain libraries" )
134+ DOMAIN_LIBRARIES = [
135+ ("torchvision==0.23.0" if use_pytorch_nightly else "torchvision" ),
136+ "torchaudio==2.8.0" if use_pytorch_nightly else "torchaudio" ,
137+ ]
138+ # Then install domain libraries
134139 subprocess .run (
135140 [
136141 sys .executable ,
137142 "-m" ,
138143 "pip" ,
139144 "install" ,
140- "-r" ,
141- "requirements-examples.txt" ,
145+ * DOMAIN_LIBRARIES ,
146+ "--extra-index-url" ,
147+ TORCH_URL ,
142148 ],
143149 check = True ,
144150 )
145151
146- print ("Installing torch domain libraries" )
147- DOMAIN_LIBRARIES = [
148- ("torchvision==0.23.0" if use_pytorch_nightly else "torchvision" ),
149- "torchaudio==2.8.0" if use_pytorch_nightly else "torchaudio" ,
150- ]
151- # Then install domain libraries
152+ print ("Installing packages in requirements-examples.txt" )
152153 subprocess .run (
153154 [
154155 sys .executable ,
155156 "-m" ,
156157 "pip" ,
157158 "install" ,
158- * DOMAIN_LIBRARIES ,
159+ "-r" ,
160+ "requirements-examples.txt" ,
159161 "--extra-index-url" ,
160162 TORCH_URL ,
161163 ],
Original file line number Diff line number Diff line change @@ -69,7 +69,7 @@ dependencies=[
6969 " sympy" ,
7070 " torch>=2.8.0,<2.9.0" ,
7171 " torchaudio>=2.8.0,<2.9.0" ,
72- " torchvisionn >=0.23.0,<0.24.0" ,
72+ " torchvision >=0.23.0,<0.24.0" ,
7373 " torchao==0.12.0" ,
7474 " tabulate" ,
7575 # See also third-party/TARGETS for buck's typing-extensions version.
You can’t perform that action at this time.
0 commit comments