update entrypoint

- when run without arguments it starts the web-interface
- can also be run with your own arguments
  - if u do so it does not start web unless u do
This commit is contained in:
mauwii 2022-10-25 00:05:52 +02:00 committed by Lincoln Stein
parent 38ff75c7ea
commit 0c3de595df

View File

@ -1,10 +1,8 @@
#!/bin/bash
set -e
cd /stable-diffusion
source "${CONDA_PREFIX}/etc/profile.d/conda.sh"
conda activate "${PROJECT_NAME}"
if [ $# -eq 0 ]; then
python3 scripts/dream.py --full_precision -o /data
# bash
else
python3 scripts/dream.py --full_precision -o /data "$@"
fi
python scripts/invoke.py \
${@:---web --host=0.0.0.0}