Skip to content

Commit

Permalink
fix(api): split up base and extras launch scripts
Browse files Browse the repository at this point in the history
  • Loading branch information
ssube committed Feb 14, 2023
1 parent 4857592 commit 4b18578
Show file tree
Hide file tree
Showing 4 changed files with 35 additions and 3 deletions.
6 changes: 6 additions & 0 deletions api/launch-extras.bat
Original file line number Diff line number Diff line change
@@ -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
28 changes: 28 additions & 0 deletions api/launch-extras.sh
Original file line number Diff line number Diff line change
@@ -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
3 changes: 1 addition & 2 deletions api/launch.bat
Original file line number Diff line number Diff line change
@@ -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
1 change: 0 additions & 1 deletion api/launch.sh
Original file line number Diff line number Diff line change
Expand Up @@ -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..."
Expand Down

0 comments on commit 4b18578

Please sign in to comment.