diff --git a/api/launch.bat b/api/launch.bat index e1bd53e8..2f20a471 100644 --- a/api/launch.bat +++ b/api/launch.bat @@ -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% diff --git a/api/launch.ps1 b/api/launch.ps1 index 4e378755..420a3554 100644 --- a/api/launch.ps1 +++ b/api/launch.ps1 @@ -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 diff --git a/api/launch.sh b/api/launch.sh index 8e793771..e4523bbc 100755 --- a/api/launch.sh +++ b/api/launch.sh @@ -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:-}