-
-
Notifications
You must be signed in to change notification settings - Fork 432
/
webui.ps1
executable file
·89 lines (68 loc) · 2.45 KB
/
webui.ps1
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
# --------------------------------------------------------------------------------------------------------------
# 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