apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): include networks in launch scripts

This commit is contained in:
Sean Sube 2024-01-14 19:50:49 -06:00
parent 513c1908a1
commit 4de917db3a
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
3 changed files with 10 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
api/launch.bat
View File

@ -5,11 +5,11 @@ echo "This launch.bat script is deprecated in favor of launch.ps1 and will be re
echo "Downloading and converting models to ONNX format..." echo "Downloading and converting models to ONNX format..."
IF "%ONNX_WEB_EXTRA_MODELS%"=="" (set ONNX_WEB_EXTRA_MODELS=..\models\extras.json) IF "%ONNX_WEB_EXTRA_MODELS%"=="" (set ONNX_WEB_EXTRA_MODELS=..\models\extras.json)
python -m onnx_web.convert ^ python -m onnx_web.convert ^
--sources ^
--diffusion ^
--upscaling ^
--correction ^ --correction ^
--diffusion ^
--networks ^ --networks ^
--sources ^
--upscaling ^
--extras=%ONNX_WEB_EXTRA_MODELS% ^ --extras=%ONNX_WEB_EXTRA_MODELS% ^
--token=%HF_TOKEN% %ONNX_WEB_EXTRA_ARGS% --token=%HF_TOKEN% %ONNX_WEB_EXTRA_ARGS%

6
api/launch.ps1
View File

@ -3,11 +3,11 @@
echo "Downloading and converting models to ONNX format..." echo "Downloading and converting models to ONNX format..."
IF ($Env:ONNX_WEB_EXTRA_MODELS -eq "") {$Env:ONNX_WEB_EXTRA_MODELS="..\models\extras.json"} IF ($Env:ONNX_WEB_EXTRA_MODELS -eq "") {$Env:ONNX_WEB_EXTRA_MODELS="..\models\extras.json"}
python -m onnx_web.convert ` python -m onnx_web.convert `
--sources `
--diffusion `
--upscaling `
--correction ` --correction `
--diffusion `
--networks ` --networks `
--sources `
--upscaling `
--extras=$Env:ONNX_WEB_EXTRA_MODELS ` --extras=$Env:ONNX_WEB_EXTRA_MODELS `
--token=$Env:HF_TOKEN $Env:ONNX_WEB_EXTRA_ARGS --token=$Env:HF_TOKEN $Env:ONNX_WEB_EXTRA_ARGS

7
api/launch.sh
View File

@ -17,10 +17,11 @@ fi
echo "Downloading and converting models to ONNX format..." echo "Downloading and converting models to ONNX format..."
python3 -m onnx_web.convert \ python3 -m onnx_web.convert \
--sources \
--diffusion \
--upscaling \
--correction \ --correction \
--diffusion \
--networks \
--sources \
--upscaling \
--extras=${ONNX_WEB_EXTRA_MODELS:-../models/extras.json} \ --extras=${ONNX_WEB_EXTRA_MODELS:-../models/extras.json} \
--token=${HF_TOKEN:-} \ --token=${HF_TOKEN:-} \
${ONNX_WEB_EXTRA_ARGS:-} ${ONNX_WEB_EXTRA_ARGS:-}