Skip to content

Commit 7d7da07

Browse files
committed
add pip fallback to stable audio
1 parent 3e7652e commit 7d7da07

File tree

1 file changed

+12
-6
lines changed

1 file changed

+12
-6
lines changed

installer_scripts/js/initializeApp.js

Lines changed: 12 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -158,15 +158,19 @@ const removeGPUChoice = () => {
158158

159159
const dry_run_flag = DEBUG_DRY_RUN ? "--dry-run " : "";
160160

161-
function pip_install(requirements, name = "") {
161+
function pip_install(requirements, name = "", pipFallback = false) {
162162
try {
163163
displayMessage(`Installing ${name || requirements} dependencies...`);
164-
$sh(`uv pip install ${dry_run_flag}${requirements} torch==${torchVersion}`);
164+
$sh(
165+
`${
166+
pipFallback ? "pip" : "uv pip"
167+
} install ${dry_run_flag}${requirements} torch==${torchVersion}`
168+
);
165169
displayMessage(
166-
`Successfully installed ${name || requirements} dependencies`
170+
`Successfully installed ${name || requirements} dependencies\n`
167171
);
168172
} catch (error) {
169-
displayMessage(`Failed to install ${name || requirements} dependencies`);
173+
displayMessage(`Failed to install ${name || requirements} dependencies\n`);
170174
}
171175
}
172176

@@ -189,7 +193,7 @@ async function pip_install_all() {
189193
pip_install("-r requirements_styletts2.txt", "StyleTTS");
190194
pip_install("-r requirements_vall_e.txt", "Vall-E-X");
191195
pip_install("-r requirements_maha_tts.txt", "Maha TTS");
192-
pip_install("-r requirements_stable_audio.txt", "Stable Audio");
196+
pip_install("-r requirements_stable_audio.txt", "Stable Audio", true);
193197
// reinstall hydra-core==1.3.2 because of fairseq
194198
pip_install("hydra-core==1.3.2", "hydra-core fix due to fairseq");
195199
pip_install("nvidia-ml-py", "nvidia-ml-py");
@@ -226,7 +230,9 @@ async function applyCondaConfig() {
226230
displayMessage(" Torch is not installed. Starting installation...\n");
227231
}
228232
} else {
229-
displayMessage(" Major version update detected. Upgrading base environment");
233+
displayMessage(
234+
" Major version update detected. Upgrading base environment"
235+
);
230236
}
231237

232238
if (fs.existsSync(gpuFile)) {

0 commit comments

Comments
 (0)