srv / sd-webui /prepare.sh
Moldwebs's picture
Update sd-webui/prepare.sh
af87a93 verified
#!/bin/bash
if [ ! -f "${ROOT}/prepare-v1.ready" ]; then
python -m pip install --upgrade pip
python -m pip install rich
python -m pip install numexpr
python -m pip install insightface
python -m pip install onnx
python -m pip install onnxruntime
python -m pip install onnxruntime-gpu
python -m pip install opencv-python
python -m pip install ifnude
python -m pip install cython
python -m pip install pytorch-lightning
python -m pip install blobfile
touch "${ROOT}/prepare-v1.ready"
fi
if [ ! -f "${ROOT}/prepare-v2.ready" ]; then
python -m pip install wandb
python -m pip install fastai==1.0.60
python -m pip install tensorboardX
python -m pip install ffmpeg
python -m pip install ffmpeg-python
python -m pip install yt-dlp
python -m pip install opencv-python
python -m pip install Pillow
touch "${ROOT}/prepare-v2.ready"
fi
if [ ! -f "${ROOT}/prepare-v3.ready" ]; then
python -m pip install scikit-learn
python -m pip install diffusers
touch "${ROOT}/prepare-v3.ready"
fi
if [ ! -f "${VOLUME_DIR_LOCAL}/styles.csv" ]; then
wget "https://huggingface.co/spaces/Moldwebs/srv/raw/main/sd-webui/styles.csv" -O ${VOLUME_DIR_LOCAL}/styles.csv
fi
if [ ! -f "${ROOT}/styles.csv" ]; then
cp ${VOLUME_DIR_LOCAL}/styles.csv ${ROOT}/styles.csv
fi