diff --git a/api/launch-extras.bat b/api/launch-extras.bat index 126df2c6..5d5ff929 100644 --- a/api/launch-extras.bat +++ b/api/launch-extras.bat @@ -1,6 +1,13 @@ echo "Downloading and converting models to ONNX format..." IF "%ONNX_WEB_EXTRA_MODELS%"=="" (set ONNX_WEB_EXTRA_MODELS=extras.json) -python -m onnx_web.convert --sources --diffusion --upscaling --correction --extras=%ONNX_WEB_EXTRA_MODELS% --token=%HF_TOKEN% +python -m onnx_web.convert ^ +--sources ^ +--diffusion ^ +--upscaling ^ +--correction ^ +--extras=%ONNX_WEB_EXTRA_MODELS% ^ +--token=%HF_TOKEN% ^ +%ONNX_WEB_EXTRA_ARGS% echo "Launching API server..." flask --app="onnx_web.main:run" run --host=0.0.0.0 diff --git a/api/launch-extras.sh b/api/launch-extras.sh index 50572aa4..5912ae87 100755 --- a/api/launch-extras.sh +++ b/api/launch-extras.sh @@ -22,7 +22,8 @@ python3 -m onnx_web.convert \ --upscaling \ --correction \ --extras=${ONNX_WEB_EXTRA_MODELS:-extras.json} \ - --token=${HF_TOKEN:-} + --token=${HF_TOKEN:-} \ + ${ONNX_WEB_EXTRA_ARGS:-} echo "Launching API server..." flask --app='onnx_web.main:run' run --host=0.0.0.0 diff --git a/api/launch.bat b/api/launch.bat index 09ddfccd..725aa0e7 100644 --- a/api/launch.bat +++ b/api/launch.bat @@ -1,5 +1,11 @@ echo "Downloading and converting models to ONNX format..." -python -m onnx_web.convert --sources --diffusion --upscaling --correction --token=%HF_TOKEN% +python -m onnx_web.convert ^ +--sources ^ +--diffusion ^ +--upscaling ^ +--correction ^ +--token=%HF_TOKEN% ^ +%ONNX_WEB_EXTRA_ARGS% echo "Launching API server..." flask --app="onnx_web.main:run" run --host=0.0.0.0 diff --git a/api/launch.sh b/api/launch.sh index 55b6ff72..bca6dfb3 100755 --- a/api/launch.sh +++ b/api/launch.sh @@ -21,7 +21,8 @@ python3 -m onnx_web.convert \ --diffusion \ --upscaling \ --correction \ - --token=${HF_TOKEN:-} + --token=${HF_TOKEN:-} \ + ${ONNX_WEB_EXTRA_ARGS:-} echo "Launching API server..." flask --app='onnx_web.main:run' run --host=0.0.0.0