InvokeAI/entrypoint.sh
2022-09-10 12:38:31 -05:00

9 lines
281 B
Bash
Executable File

#!/bin/bash
ln -sf /data/sd-v1-4.ckpt /stable-diffusion/models/ldm/stable-diffusion-v1/model.ckpt
if [ $# -eq 0 ]; then
python3 /stable-diffusion/scripts/dream.py --full_precision -o /data
else
python3 /stable-diffusion/scripts/dream.py --full_precision -o /data "$@"
fi