Spaces:
Runtime error
Runtime error
File size: 2,510 Bytes
c19ca42 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 |
# --------------------------------------------------------------------------------------------------------------
# Do not make any changes to this file, change the variables in webui-user.ps1 instead and call this file
# --------------------------------------------------------------------------------------------------------------
function ShowStdOutStdErr {
Write-Output "exit code: $LASTEXITCODE"
if ((Get-Item tmp\stdout.txt).Length -ne 0) {
Write-Output "`nstdout:"
Get-Content tmp\stdout.txt
}
if ((Get-Item tmp\stderr.txt).Length -ne 0) {
Write-Error "`nstderr:"
Get-Content tmp\stderr.txt
}
Write-Output "`nLaunch Failed"
Pause
}
$PYTHON = if ($env:PYTHON) { $env:PYTHON } else { 'python' }
$VENV_DIR = if ($env:VENV_DIR) { $env:VENV_DIR } else { Join-Path $PSScriptRoot 'venv' }
New-Item -Path 'tmp' -ItemType Directory -ErrorAction SilentlyContinue
try {
& $PYTHON -c '' 2>tmp\stderr.txt | Out-File tmp\stdout.txt
} catch {
Write-Output 'Cannot launch python'
. ShowStdOutStdErr
exit
}
try {
& $PYTHON -m pip --help 2>tmp\stderr.txt | Out-File tmp\stdout.txt
} catch {
if (!$env:PIP_INSTALLER_LOCATION) {
. ShowStdOutStdErr
exit
}
& $PYTHON $env:PIP_INSTALLER_LOCATION 2>tmp\stderr.txt | Out-File tmp\stdout.txt
if ($LASTEXITCODE -ne 0) {
Write-Output 'Cannot install pip'
. ShowStdOutStdErr
exit
}
}
if ($VENV_DIR -ne '-' -and $env:SKIP_VENV -ne '1') {
if (Test-Path (Join-Path $VENV_DIR 'Scripts\Python.exe')) {
$PYTHON = Join-Path $VENV_DIR 'Scripts\Python.exe'
} else {
$PYTHON_FULLNAME = & $PYTHON -c 'import sys; print(sys.executable)'
Write-Output "Using python: $PYTHON_FULLNAME"
Write-Output "Creating VENV: $VENV_DIR"
& $PYTHON_FULLNAME -m venv $VENV_DIR 2>tmp\stderr.txt | Out-File tmp\stdout.txt
if ($LASTEXITCODE -eq 0) {
$PYTHON = Join-Path $VENV_DIR 'Scripts\Python.exe'
} else {
Write-Output "Failed creating VENV: '$VENV_DIR'"
. ShowStdOutStdErr
exit
}
}
Write-Output "Using VENV: $VENV_DIR"
}
if ($env:ACCELERATE -eq 'True') {
$ACCELERATE = Join-Path $VENV_DIR 'Scripts\accelerate.exe'
if (Test-Path $ACCELERATE) {
Write-Output 'Using accelerate'
& $ACCELERATE launch --num_cpu_threads_per_process=6 launch.py $args
}
} else {
& $PYTHON launch.py $args
}
Pause
|