Add support and enable deepdanbooru setting

This commit is contained in:
Imaginator 2022-10-17 13:15:25 -04:00
parent 31e4dec08f
commit ef7e5d7bfb
2 changed files with 7 additions and 1 deletions

View file

@ -36,7 +36,7 @@ services:
profiles: ["auto"]
build: ./services/AUTOMATIC1111
environment:
- CLI_ARGS=--allow-code --medvram --xformers
- CLI_ARGS=--allow-code --medvram --xformers --deepdanbooru
auto-cpu:
<<: *automatic

View file

@ -73,6 +73,12 @@ RUN pip install --prefer-binary --no-cache-dir opencv-python-headless \
COPY --from=xformers /wheel.whl xformers-0.0.14.dev0-cp310-cp310-linux_x86_64.whl
RUN pip install xformers-0.0.14.dev0-cp310-cp310-linux_x86_64.whl
RUN <<EOF
pip install tensorflow
pip install tensorflow-io
pip install git+https://github.com/KichangKim/DeepDanbooru.git@edf73df4cdaeea2cf00e9ac08bd8a9026b7a7b26#egg=deepdanbooru deepdanbooru
EOF
COPY . /docker
RUN <<EOF
chmod +x /docker/mount.sh && python3 /docker/info.py ${ROOT}/modules/ui.py