@@ -125,17 +125,14 @@ await PrerequisiteHelper
125125 await using var venvRunner = await SetupVenvPure ( installLocation ) . ConfigureAwait ( false ) ;
126126
127127 // Extra dep needed before running setup since v23.0.x
128- var pipArgs = new PipInstallArgs ( "rich" , "packaging" ) ;
128+ var pipArgs = new PipInstallArgs ( "rich" , "packaging" , "setuptools" , "uv" ) ;
129129 if ( installedPackage . PipOverrides != null )
130130 {
131131 pipArgs = pipArgs . WithUserOverrides ( installedPackage . PipOverrides ) ;
132132 }
133133
134134 await venvRunner . PipInstall ( pipArgs ) . ConfigureAwait ( false ) ;
135135
136- < << << << HEAD
137- if ( Compat . IsWindows )
138- == == == =
139136 var isLegacyNvidia =
140137 SettingsManager . Settings . PreferredGpu ? . IsLegacyNvidiaGpu ( ) ?? HardwareHelper . HasLegacyNvidiaGpu ( ) ;
141138 var torchExtraIndex = isLegacyNvidia ? "cu126" : "cu128" ;
@@ -150,68 +147,44 @@ await PrerequisiteHelper
150147 . AddArg ( "--force-reinstall" ) ;
151148
152149 if ( installedPackage . PipOverrides != null )
153- > >>> >>> c422920b ( Merge pull request #1076 from ionite34/ fix- legacy - gpu )
154150 {
155- await venvRunner
156- . CustomInstall ( [ "setup/setup_windows.py" , "--headless" ] , onConsoleOutput )
157- . ConfigureAwait ( false ) ;
151+ pipArgs = pipArgs . WithUserOverrides ( installedPackage . PipOverrides ) ;
158152 }
159- else if ( Compat . IsLinux )
153+
154+ await venvRunner . PipInstall ( pipArgs , onConsoleOutput ) . ConfigureAwait ( false ) ;
155+
156+ if ( Compat . IsLinux )
160157 {
161158 await venvRunner
162159 . CustomInstall (
163160 [
164161 "setup/setup_linux.py" ,
165162 "--platform-requirements-file=requirements_linux.txt" ,
166- "--no_run_accelerate"
163+ "--no_run_accelerate" ,
167164 ] ,
168165 onConsoleOutput
169166 )
170167 . ConfigureAwait ( false ) ;
168+ pipArgs = new PipInstallArgs ( ) ;
171169 }
172-
173- var isBlackwell =
174- SettingsManager . Settings . PreferredGpu ? . IsBlackwellGpu ( ) ?? HardwareHelper . HasBlackwellGpu ( ) ;
175-
176- if ( isBlackwell )
170+ else if ( Compat . IsWindows )
177171 {
172+ var requirements = new FilePath ( installLocation , "requirements_windows.txt" ) ;
178173 pipArgs = new PipInstallArgs ( )
179- . WithTorch ( )
180- . WithTorchVision ( )
181- . WithTorchAudio ( )
182- . WithTorchExtraIndex ( "cu128" )
183- . AddArg ( "--force-reinstall" ) ;
184-
185- if ( installedPackage . PipOverrides != null )
186- {
187- pipArgs = pipArgs . WithUserOverrides ( installedPackage . PipOverrides ) ;
188- }
189-
190- await venvRunner . PipInstall ( pipArgs , onConsoleOutput ) . ConfigureAwait ( false ) ;
191-
192- pipArgs = new PipInstallArgs ( )
193- . AddArg ( "--pre" )
194- . AddArg ( "-U" )
195- . AddArg ( "--no-deps" )
196- . AddArg ( "xformers" ) ;
197-
198- if ( installedPackage . PipOverrides != null )
199- {
200- pipArgs = pipArgs . WithUserOverrides ( installedPackage . PipOverrides ) ;
201- }
202-
203- await venvRunner . PipInstall ( pipArgs , onConsoleOutput ) . ConfigureAwait ( false ) ;
204-
205- pipArgs = new PipInstallArgs ( ) . AddArg ( "-U" ) . AddArg ( "bitsandbytes" ) ;
206-
207- if ( installedPackage . PipOverrides != null )
208- {
209- pipArgs = pipArgs . WithUserOverrides ( installedPackage . PipOverrides ) ;
210- }
174+ . WithParsedFromRequirementsTxt (
175+ await requirements . ReadAllTextAsync ( cancellationToken ) . ConfigureAwait ( false ) ,
176+ "bitsandbytes==0\\ .44\\ .0"
177+ )
178+ . AddArg ( "bitsandbytes" ) ;
179+ }
211180
212- await venvRunner . PipInstall ( pipArgs , onConsoleOutput ) . ConfigureAwait ( false ) ;
213- await venvRunner . PipInstall ( "numpy==1.26.4" , onConsoleOutput ) . ConfigureAwait ( false ) ;
181+ if ( installedPackage . PipOverrides != null )
182+ {
183+ pipArgs = pipArgs . WithUserOverrides ( installedPackage . PipOverrides ) ;
214184 }
185+
186+ await venvRunner . PipInstall ( pipArgs , onConsoleOutput ) . ConfigureAwait ( false ) ;
187+ await venvRunner . PipInstall ( "numpy==1.26.4" , onConsoleOutput ) . ConfigureAwait ( false ) ;
215188 }
216189
217190 public override async Task RunPackage (
0 commit comments