Deepdanbooru Support (#194)

Builds on top of #150 

Thanks to @pirahtays

Co-authored-by: Imaginator <mriegel@gmail.com>
Co-authored-by: pirahtays <35934562+pirahtays@users.noreply.github.com>
This commit is contained in:
AbdBarho
2022-11-04 22:41:38 +01:00
committed by GitHub
parent 1cfb915d12
commit 5450583be1
4 changed files with 6 additions and 2 deletions

View File

@@ -59,6 +59,9 @@ RUN mkdir ${ROOT}/interrogate && cp ${ROOT}/repositories/clip-interrogator/data/
RUN pip install --prefer-binary --no-cache-dir -r ${ROOT}/repositories/CodeFormer/requirements.txt
ARG DEEPDANBOORU="0"
RUN [[ "${DEEPDANBOORU:-0}" == "0" ]] && : || pip install tensorflow-cpu==2.10 tensorflow-io==0.27.0 git+https://github.com/KichangKim/DeepDanbooru.git@edf73df4cdaeea2cf00e9ac08bd8a9026b7a7b26#egg=deepdanbooru
# Note: don't update the sha of previous versions because the install will take forever
# instead, update the repo state in a later step