fix(api): split up base and extras launch scripts
This commit is contained in:
parent
4857592dba
commit
4b1857883f
|
@ -0,0 +1,6 @@
|
|||
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%
|
||||
|
||||
echo "Launching API server..."
|
||||
flask --app=onnx_web.serve run --host=0.0.0.0
|
|
@ -0,0 +1,28 @@
|
|||
#! /bin/sh
|
||||
|
||||
set -eu
|
||||
|
||||
if [ -n "${VIRTUAL_ENV+set}" ]; then
|
||||
echo "Using current virtual env..."
|
||||
else
|
||||
if [ -d "onnx_env" ]; then
|
||||
echo "Loading existing virtual env..."
|
||||
. onnx_env/bin/activate
|
||||
else
|
||||
echo "Creating new virtual env..."
|
||||
python -m venv onnx_env
|
||||
. onnx_env/bin/activate
|
||||
fi
|
||||
fi
|
||||
|
||||
echo "Downloading and converting models to ONNX format..."
|
||||
python3 -m onnx_web.convert \
|
||||
--sources \
|
||||
--diffusion \
|
||||
--upscaling \
|
||||
--correction \
|
||||
--extras=${ONNX_WEB_EXTRA_MODELS:-extras.json} \
|
||||
--token=${HF_TOKEN:-}
|
||||
|
||||
echo "Launching API server..."
|
||||
flask --app=onnx_web.serve run --host=0.0.0.0
|
|
@ -1,6 +1,5 @@
|
|||
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=extras.json --token=%HF_TOKEN%
|
||||
python -m onnx_web.convert --sources --diffusion --upscaling --correction --token=%HF_TOKEN%
|
||||
|
||||
echo "Launching API server..."
|
||||
flask --app=onnx_web.serve run --host=0.0.0.0
|
||||
|
|
|
@ -21,7 +21,6 @@ python3 -m onnx_web.convert \
|
|||
--diffusion \
|
||||
--upscaling \
|
||||
--correction \
|
||||
--extras=${ONNX_WEB_EXTRA_MODELS:-extras.json} \
|
||||
--token=${HF_TOKEN:-}
|
||||
|
||||
echo "Launching API server..."
|
||||
|
|
Loading…
Reference in New Issue