1
0
Fork 0

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

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..."
IF "%ONNX_WEB_EXTRA_MODELS%"=="" (set ONNX_WEB_EXTRA_MODELS=..\models\extras.json)
python -m onnx_web.convert ^
--sources ^
--diffusion ^
--upscaling ^
--correction ^
--diffusion ^
--networks ^
--sources ^
--upscaling ^
--extras=%ONNX_WEB_EXTRA_MODELS% ^
--token=%HF_TOKEN% %ONNX_WEB_EXTRA_ARGS%

View File

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

View File

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