mirror of
https://github.com/AbdBarho/stable-diffusion-webui-docker.git
synced 2026-02-16 04:24:20 +01:00
Compare commits
No commits in common. "master" and "3.1.0" have entirely different histories.
|
|
@ -3,4 +3,3 @@
|
||||||
set -Eeuo pipefail
|
set -Eeuo pipefail
|
||||||
|
|
||||||
find services -name "*.sh" -exec git update-index --chmod=+x {} \;
|
find services -name "*.sh" -exec git update-index --chmod=+x {} \;
|
||||||
find .devscripts -name "*.sh" -exec git update-index --chmod=+x {} \;
|
|
||||||
|
|
|
||||||
|
|
@ -1,9 +0,0 @@
|
||||||
#!/bin/bash
|
|
||||||
|
|
||||||
set -Eeuo pipefail
|
|
||||||
|
|
||||||
echo "Moving everything in output to output/old..."
|
|
||||||
mv output old
|
|
||||||
mkdir output
|
|
||||||
mv old/.gitignore output
|
|
||||||
mv old output
|
|
||||||
|
|
@ -1,28 +0,0 @@
|
||||||
#!/bin/bash
|
|
||||||
|
|
||||||
set -Eeuo pipefail
|
|
||||||
|
|
||||||
echo "Renaming..."
|
|
||||||
|
|
||||||
# compatible with default auto-names
|
|
||||||
mv -v ./data/StableDiffusion ./data/Stable-diffusion
|
|
||||||
mv -v ./data/Deepdanbooru ./data/torch_deepdanbooru
|
|
||||||
|
|
||||||
# casing problem on windows
|
|
||||||
mv -v ./data/Hypernetworks ./data/hypernetworks1
|
|
||||||
mv -v ./data/hypernetworks1 ./data/hypernetworks
|
|
||||||
|
|
||||||
mv -v ./data/MiDaS ./data/midas1
|
|
||||||
mv -v ./data/midas1 ./data/midas
|
|
||||||
|
|
||||||
|
|
||||||
echo "Moving folders..."
|
|
||||||
|
|
||||||
mkdir -pv ./final
|
|
||||||
|
|
||||||
mv -v ./data/config ./final/config
|
|
||||||
mv -v ./data/.cache ./final/.cache
|
|
||||||
mv -v ./data/embeddings ./final/embeddings
|
|
||||||
mv -v ./data ./final/models
|
|
||||||
|
|
||||||
mv -v ./final ./data
|
|
||||||
28
.github/ISSUE_TEMPLATE/bug.md
vendored
28
.github/ISSUE_TEMPLATE/bug.md
vendored
|
|
@ -1,9 +1,10 @@
|
||||||
---
|
---
|
||||||
name: Bug
|
name: Bug
|
||||||
about: Report a bug
|
about: Report a bug
|
||||||
title: ""
|
title: ''
|
||||||
labels: bug
|
labels: bug
|
||||||
assignees: ""
|
assignees: ''
|
||||||
|
|
||||||
---
|
---
|
||||||
|
|
||||||
<!-- PLEASE FILL THIS OUT, IT WILL MAKE BOTH OF OUR LIVES EASIER -->
|
<!-- PLEASE FILL THIS OUT, IT WILL MAKE BOTH OF OUR LIVES EASIER -->
|
||||||
|
|
@ -13,27 +14,28 @@ assignees: ""
|
||||||
- [ ] It is not in the [FAQ](https://github.com/AbdBarho/stable-diffusion-webui-docker/wiki/FAQ), I checked.
|
- [ ] It is not in the [FAQ](https://github.com/AbdBarho/stable-diffusion-webui-docker/wiki/FAQ), I checked.
|
||||||
- [ ] It is not in the [issues](https://github.com/AbdBarho/stable-diffusion-webui-docker/issues?q=), I searched.
|
- [ ] It is not in the [issues](https://github.com/AbdBarho/stable-diffusion-webui-docker/issues?q=), I searched.
|
||||||
|
|
||||||
|
|
||||||
**Describe the bug**
|
**Describe the bug**
|
||||||
|
|
||||||
<!-- tried to run the app, my cat exploded -->
|
<!-- tried to run the app, my cat exploded -->
|
||||||
|
|
||||||
|
|
||||||
**Which UI**
|
**Which UI**
|
||||||
|
|
||||||
auto or auto-cpu or invoke or comfy?
|
hlky or auto or auto-cpu or lstein?
|
||||||
|
|
||||||
|
|
||||||
**Hardware / Software**
|
**Hardware / Software**
|
||||||
|
- OS: [e.g. Windows 10 / Ubuntu ]
|
||||||
- OS: [e.g. Windows 10 / Ubuntu 22.04]
|
- OS version: <!-- on windows, use the command `winver` to find out, on ubuntu `lsb_release -d` -->
|
||||||
- OS version: <!-- on windows, use the command `winver` to find out, on ubuntu `lsb_release -d` -->
|
- WSL version (if applicable): <!-- get using `wsl -l -v` -->
|
||||||
- WSL version (if applicable): <!-- get using `wsl -l -v` -->
|
- Docker Version: <!-- get using `docker version` -->
|
||||||
- Docker Version: <!-- get using `docker version` -->
|
- Docker compose version: <!-- get using `docker compose version` -->
|
||||||
- Docker compose version: <!-- get using `docker compose version` -->
|
- Repo version: <!-- tag, commit sha, or "from master" -->
|
||||||
- Repo version: <!-- tag, commit sha, or "from master" -->
|
- RAM:
|
||||||
- RAM:
|
- GPU/VRAM:
|
||||||
- GPU/VRAM:
|
|
||||||
|
|
||||||
**Steps to Reproduce**
|
**Steps to Reproduce**
|
||||||
|
|
||||||
1. Go to '...'
|
1. Go to '...'
|
||||||
2. Click on '....'
|
2. Click on '....'
|
||||||
3. Scroll down to '....'
|
3. Scroll down to '....'
|
||||||
|
|
|
||||||
5
.github/ISSUE_TEMPLATE/config.yml
vendored
5
.github/ISSUE_TEMPLATE/config.yml
vendored
|
|
@ -1,5 +0,0 @@
|
||||||
blank_issues_enabled: false
|
|
||||||
contact_links:
|
|
||||||
- name: Feature request? Questions regarding some extension?
|
|
||||||
url: https://github.com/AbdBarho/stable-diffusion-webui-docker/discussions
|
|
||||||
about: Please use the discussions tab
|
|
||||||
4
.github/pull_request_template.md
vendored
4
.github/pull_request_template.md
vendored
|
|
@ -9,5 +9,5 @@ Closes issue #
|
||||||
### Update versions
|
### Update versions
|
||||||
|
|
||||||
- auto: https://github.com/AUTOMATIC1111/stable-diffusion-webui/commit/
|
- auto: https://github.com/AUTOMATIC1111/stable-diffusion-webui/commit/
|
||||||
- invoke: https://github.com/invoke-ai/InvokeAI/commit/
|
- hlky: https://github.com/sd-webui/stable-diffusion-webui/commit/
|
||||||
- comfy: https://github.com/comfyanonymous/ComfyUI/commit/
|
- lstein: https://github.com/invoke-ai/InvokeAI/commit/
|
||||||
|
|
|
||||||
4
.github/workflows/docker.yml
vendored
4
.github/workflows/docker.yml
vendored
|
|
@ -14,10 +14,12 @@ jobs:
|
||||||
matrix:
|
matrix:
|
||||||
profile:
|
profile:
|
||||||
- auto
|
- auto
|
||||||
- comfy
|
- hlky
|
||||||
|
- lstein
|
||||||
- download
|
- download
|
||||||
runs-on: ubuntu-latest
|
runs-on: ubuntu-latest
|
||||||
name: ${{ matrix.profile }}
|
name: ${{ matrix.profile }}
|
||||||
steps:
|
steps:
|
||||||
- uses: actions/checkout@v3
|
- uses: actions/checkout@v3
|
||||||
|
# better caching?
|
||||||
- run: docker compose --profile ${{ matrix.profile }} build --progress plain
|
- run: docker compose --profile ${{ matrix.profile }} build --progress plain
|
||||||
|
|
|
||||||
4
.github/workflows/stale.yml
vendored
4
.github/workflows/stale.yml
vendored
|
|
@ -1,13 +1,13 @@
|
||||||
name: 'Close stale issues and PRs'
|
name: 'Close stale issues and PRs'
|
||||||
on:
|
on:
|
||||||
schedule:
|
schedule:
|
||||||
- cron: '0 0 * * *'
|
- cron: '30 1 * * *'
|
||||||
|
|
||||||
jobs:
|
jobs:
|
||||||
stale:
|
stale:
|
||||||
runs-on: ubuntu-latest
|
runs-on: ubuntu-latest
|
||||||
steps:
|
steps:
|
||||||
- uses: actions/stale@v6
|
- uses: actions/stale@v5
|
||||||
with:
|
with:
|
||||||
only-labels: awaiting-response
|
only-labels: awaiting-response
|
||||||
stale-issue-message: This issue is stale because it has been open 14 days with no activity. Remove stale label or comment or this will be closed in 7 days.
|
stale-issue-message: This issue is stale because it has been open 14 days with no activity. Remove stale label or comment or this will be closed in 7 days.
|
||||||
|
|
|
||||||
36
.github/workflows/xformers.yml
vendored
Normal file
36
.github/workflows/xformers.yml
vendored
Normal file
|
|
@ -0,0 +1,36 @@
|
||||||
|
name: Build Xformers
|
||||||
|
|
||||||
|
on:
|
||||||
|
workflow_dispatch: {}
|
||||||
|
|
||||||
|
jobs:
|
||||||
|
build:
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
timeout-minutes: 180
|
||||||
|
container:
|
||||||
|
image: python:3.10-slim
|
||||||
|
env:
|
||||||
|
DEBIAN_FRONTEND: noninteractive
|
||||||
|
XFORMERS_DISABLE_FLASH_ATTN: 1
|
||||||
|
FORCE_CUDA: 1
|
||||||
|
TORCH_CUDA_ARCH_LIST: "6.0;6.1;6.2;7.0;7.2;7.5;8.0;8.6"
|
||||||
|
NVCC_FLAGS: --use_fast_math -DXFORMERS_MEM_EFF_ATTENTION_DISABLE_BACKWARD
|
||||||
|
MAX_JOBS: 4
|
||||||
|
steps:
|
||||||
|
- run: |
|
||||||
|
apt-get update
|
||||||
|
apt-get install gpg wget git -y
|
||||||
|
wget https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64/cuda-keyring_1.0-1_all.deb
|
||||||
|
dpkg -i cuda-keyring_1.0-1_all.deb
|
||||||
|
apt-get update
|
||||||
|
apt-get install cuda-nvcc-11-8 cuda-libraries-dev-11-8 -y
|
||||||
|
|
||||||
|
export PIP_CACHE_DIR=$(pwd)/cache
|
||||||
|
pip install ninja install torch --extra-index-url https://download.pytorch.org/whl/cu113
|
||||||
|
|
||||||
|
pip wheel --wheel-dir=data git+https://github.com/facebookresearch/xformers.git@3633e1afc7bffbe61957f04e7bb1a742ee910ace#egg=xformers
|
||||||
|
- name: Artifacts
|
||||||
|
uses: actions/upload-artifact@v3
|
||||||
|
with:
|
||||||
|
name: xformers
|
||||||
|
path: data/xformers-0.0.14.dev0-cp310-cp310-linux_x86_64.whl
|
||||||
4
.gitignore
vendored
4
.gitignore
vendored
|
|
@ -1,6 +1,2 @@
|
||||||
/.devcontainer
|
/.devcontainer
|
||||||
/docker-compose.override.yml
|
/docker-compose.override.yml
|
||||||
|
|
||||||
# VSCode specific
|
|
||||||
*.code-workspace
|
|
||||||
/.vscode
|
|
||||||
|
|
|
||||||
13
LICENSE
13
LICENSE
|
|
@ -87,3 +87,16 @@ processes, such as predicting an individual will commit fraud/crime
|
||||||
commitment (e.g. by text profiling, drawing causal relationships between
|
commitment (e.g. by text profiling, drawing causal relationships between
|
||||||
assertions made in documents, indiscriminate and arbitrarily-targeted
|
assertions made in documents, indiscriminate and arbitrarily-targeted
|
||||||
use).
|
use).
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
By using this software, you also agree to the following licenses:
|
||||||
|
https://github.com/CompVis/stable-diffusion/blob/main/LICENSE
|
||||||
|
https://github.com/sd-webui/stable-diffusion-webui/blob/master/LICENSE
|
||||||
|
https://github.com/invoke-ai/InvokeAI/blob/main/LICENSE
|
||||||
|
https://github.com/cszn/BSRGAN/blob/main/LICENSE
|
||||||
|
https://github.com/sczhou/CodeFormer/blob/master/LICENSE
|
||||||
|
https://github.com/TencentARC/GFPGAN/blob/master/LICENSE
|
||||||
|
https://github.com/xinntao/Real-ESRGAN/blob/master/LICENSE
|
||||||
|
https://github.com/xinntao/ESRGAN/blob/master/LICENSE
|
||||||
|
https://github.com/cszn/SCUNet/blob/main/LICENSE
|
||||||
|
|
|
||||||
68
README.md
68
README.md
|
|
@ -2,33 +2,68 @@
|
||||||
|
|
||||||
Run Stable Diffusion on your machine with a nice UI without any hassle!
|
Run Stable Diffusion on your machine with a nice UI without any hassle!
|
||||||
|
|
||||||
## Setup & Usage
|
This repository provides multiple UIs for you to play around with stable diffusion:
|
||||||
|
|
||||||
Visit the wiki for [Setup](https://github.com/AbdBarho/stable-diffusion-webui-docker/wiki/Setup) and [Usage](https://github.com/AbdBarho/stable-diffusion-webui-docker/wiki/Usage) instructions, checkout the [FAQ](https://github.com/AbdBarho/stable-diffusion-webui-docker/wiki/FAQ) page if you face any problems, or create a new issue!
|
|
||||||
|
|
||||||
## Features
|
## Features
|
||||||
|
|
||||||
This repository provides multiple UIs for you to play around with stable diffusion:
|
### AUTOMATIC1111
|
||||||
|
|
||||||
### [AUTOMATIC1111](https://github.com/AUTOMATIC1111/stable-diffusion-webui)
|
[AUTOMATIC1111's fork](https://github.com/AUTOMATIC1111/stable-diffusion-webui) is imho the most feature rich yet elegant UI:
|
||||||
|
|
||||||
[Full feature list here](https://github.com/AUTOMATIC1111/stable-diffusion-webui-feature-showcase), Screenshots:
|
- Text to image, with many samplers and even negative prompts!
|
||||||
|
- Image to image, with masking, cropping, in-painting, out-painting, variations.
|
||||||
|
- GFPGAN, RealESRGAN, LDSR, CodeFormer.
|
||||||
|
- Loopback, prompt weighting, prompt matrix, X/Y plot
|
||||||
|
- Live preview of the generated images.
|
||||||
|
- Highly optimized 4GB GPU support, or even CPU only!
|
||||||
|
- Textual inversion allows you to use pretrained textual inversion embeddings
|
||||||
|
- [Full feature list here](https://github.com/AUTOMATIC1111/stable-diffusion-webui-feature-showcase)
|
||||||
|
|
||||||
| Text to image | Image to image | Extras |
|
| Text to image | Image to image | Extras |
|
||||||
| ---------------------------------------------------------------------------------------------------------- | ---------------------------------------------------------------------------------------------------------- | ---------------------------------------------------------------------------------------------------------- |
|
| ---------------------------------------------------------------------------------------------------------- | ---------------------------------------------------------------------------------------------------------- | ---------------------------------------------------------------------------------------------------------- |
|
||||||
|  |  |  |
|
|  |  |  |
|
||||||
|
|
||||||
### [ComfyUI](https://github.com/comfyanonymous/ComfyUI)
|
### hlky (sd-webui / sygil-webui)
|
||||||
|
|
||||||
[Full feature list here](https://github.com/comfyanonymous/ComfyUI#features), Screenshot:
|
[hlky's fork](https://github.com/Sygil-Dev/sygil-webui) is one of the most popular UIs, with many features:
|
||||||
|
|
||||||
| Workflow |
|
- Text to image, with many samplers
|
||||||
| -------------------------------------------------------------------------------- |
|
- Image to image, with masking, cropping, in-painting, variations.
|
||||||
|  |
|
- GFPGAN, RealESRGAN, LDSR, GoBig, GoLatent
|
||||||
|
- Loopback, prompt weighting
|
||||||
|
- 6GB or even 4GB GPU support!
|
||||||
|
- [Full feature list here](https://github.com/Sygil-Dev/sygil-webui/blob/master/README.md)
|
||||||
|
|
||||||
|
Screenshots:
|
||||||
|
|
||||||
|
| Text to image | Image to image | Image Lab |
|
||||||
|
| ---------------------------------------------------------------------------------------------------------- | ---------------------------------------------------------------------------------------------------------- | ---------------------------------------------------------------------------------------------------------- |
|
||||||
|
|  |  |  |
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
### lstein (InvokeAI)
|
||||||
|
|
||||||
|
[lstein's fork](https://github.com/invoke-ai/InvokeAI) is one of the earliest with a wonderful WebUI.
|
||||||
|
- Text to image, with many samplers
|
||||||
|
- Image to image
|
||||||
|
- 4GB GPU support
|
||||||
|
- More coming!
|
||||||
|
- [Full feature list here](https://github.com/invoke-ai/InvokeAI#features)
|
||||||
|
|
||||||
|
| Text to image | Image to image | Extras |
|
||||||
|
| ---------------------------------------------------------------------------------------------------------- | ---------------------------------------------------------------------------------------------------------- | ---------------------------------------------------------------------------------------------------------- |
|
||||||
|
|  |  |  |
|
||||||
|
|
||||||
|
## Setup & Usage
|
||||||
|
|
||||||
|
Visit the wiki for [Setup](https://github.com/AbdBarho/stable-diffusion-webui-docker/wiki/Setup) and [Usage](https://github.com/AbdBarho/stable-diffusion-webui-docker/wiki/Usage) instructions, checkout the [FAQ](https://github.com/AbdBarho/stable-diffusion-webui-docker/wiki/FAQ) page if you face any problems, or create a new issue!
|
||||||
|
|
||||||
## Contributing
|
## Contributing
|
||||||
|
|
||||||
Contributions are welcome! **Create a discussion first of what the problem is and what you want to contribute (before you implement anything)**
|
Contributions are welcome!
|
||||||
|
|
||||||
|
### **Create an issue first of what you want to contribute (before you implement anything)**
|
||||||
|
|
||||||
## Disclaimer
|
## Disclaimer
|
||||||
|
|
||||||
|
|
@ -40,9 +75,10 @@ This license of this software forbids you from sharing any content that violates
|
||||||
|
|
||||||
Special thanks to everyone behind these awesome projects, without them, none of this would have been possible:
|
Special thanks to everyone behind these awesome projects, without them, none of this would have been possible:
|
||||||
|
|
||||||
|
- [hlky/stable-diffusion-webui](https://github.com/hlky/stable-diffusion-webui)
|
||||||
- [AUTOMATIC1111/stable-diffusion-webui](https://github.com/AUTOMATIC1111/stable-diffusion-webui)
|
- [AUTOMATIC1111/stable-diffusion-webui](https://github.com/AUTOMATIC1111/stable-diffusion-webui)
|
||||||
- [InvokeAI](https://github.com/invoke-ai/InvokeAI)
|
- [lstein/stable-diffusion](https://github.com/lstein/stable-diffusion)
|
||||||
- [ComfyUI](https://github.com/comfyanonymous/ComfyUI)
|
|
||||||
- [CompVis/stable-diffusion](https://github.com/CompVis/stable-diffusion)
|
- [CompVis/stable-diffusion](https://github.com/CompVis/stable-diffusion)
|
||||||
- [Sygil-webui](https://github.com/Sygil-Dev/sygil-webui)
|
- [hlky/sd-enable-textual-inversion](https://github.com/hlky/sd-enable-textual-inversion)
|
||||||
- and many many more.
|
- [devilismyfriend/latent-diffusion](https://github.com/devilismyfriend/latent-diffusion)
|
||||||
|
- [Hafiidz/latent-diffusion](https://github.com/Hafiidz/latent-diffusion)
|
||||||
|
|
|
||||||
15
data/.gitignore
vendored
15
data/.gitignore
vendored
|
|
@ -1,4 +1,17 @@
|
||||||
|
# for all of the stuff downloaded by transformers, pytorch, and others
|
||||||
/.cache
|
/.cache
|
||||||
|
# for UIs
|
||||||
/config
|
/config
|
||||||
|
# for all stable diffusion models (main, waifu diffusion, etc..)
|
||||||
|
/StableDiffusion
|
||||||
|
# others
|
||||||
|
/Codeformer
|
||||||
|
/GFPGAN
|
||||||
|
/ESRGAN
|
||||||
|
/BSRGAN
|
||||||
|
/RealESRGAN
|
||||||
|
/SwinIR
|
||||||
|
/ScuNET
|
||||||
|
/LDSR
|
||||||
|
/Hypernetworks
|
||||||
/embeddings
|
/embeddings
|
||||||
/models
|
|
||||||
|
|
|
||||||
|
|
@ -1,18 +1,18 @@
|
||||||
|
version: '3.9'
|
||||||
|
|
||||||
x-base_service: &base_service
|
x-base_service: &base_service
|
||||||
ports:
|
ports:
|
||||||
- "${WEBUI_PORT:-7860}:7860"
|
- "7860:7860"
|
||||||
volumes:
|
volumes:
|
||||||
- &v1 ./data:/data
|
- &v1 ./data:/data
|
||||||
- &v2 ./output:/output
|
- &v2 ./output:/output
|
||||||
stop_signal: SIGKILL
|
|
||||||
tty: true
|
|
||||||
deploy:
|
deploy:
|
||||||
resources:
|
resources:
|
||||||
reservations:
|
reservations:
|
||||||
devices:
|
devices:
|
||||||
- driver: nvidia
|
- driver: nvidia
|
||||||
device_ids: ['0']
|
device_ids: ['0']
|
||||||
capabilities: [compute, utility]
|
capabilities: [gpu]
|
||||||
|
|
||||||
name: webui-docker
|
name: webui-docker
|
||||||
|
|
||||||
|
|
@ -23,33 +23,35 @@ services:
|
||||||
volumes:
|
volumes:
|
||||||
- *v1
|
- *v1
|
||||||
|
|
||||||
|
hlky:
|
||||||
|
<<: *base_service
|
||||||
|
profiles: ["hlky"]
|
||||||
|
build: ./services/hlky/
|
||||||
|
image: sd-hlky:4
|
||||||
|
environment:
|
||||||
|
- CLI_ARGS=--optimized-turbo
|
||||||
|
- USE_STREAMLIT=0
|
||||||
|
|
||||||
auto: &automatic
|
auto: &automatic
|
||||||
<<: *base_service
|
<<: *base_service
|
||||||
profiles: ["auto"]
|
profiles: ["auto"]
|
||||||
build: ./services/AUTOMATIC1111
|
build: ./services/AUTOMATIC1111
|
||||||
image: sd-auto:78
|
image: sd-auto:7
|
||||||
environment:
|
environment:
|
||||||
- CLI_ARGS=--allow-code --medvram --xformers --enable-insecure-extension-access --api
|
- CLI_ARGS=--allow-code --medvram --xformers
|
||||||
|
|
||||||
auto-cpu:
|
auto-cpu:
|
||||||
<<: *automatic
|
<<: *automatic
|
||||||
profiles: ["auto-cpu"]
|
profiles: ["auto-cpu"]
|
||||||
deploy: {}
|
deploy: {}
|
||||||
environment:
|
environment:
|
||||||
- CLI_ARGS=--no-half --precision full --allow-code --enable-insecure-extension-access --api
|
- CLI_ARGS=--no-half --precision full
|
||||||
|
|
||||||
comfy: &comfy
|
lstein:
|
||||||
<<: *base_service
|
<<: *base_service
|
||||||
profiles: ["comfy"]
|
profiles: ["lstein"]
|
||||||
build: ./services/comfy/
|
build: ./services/lstein/
|
||||||
image: sd-comfy:7
|
image: sd-lstein:4
|
||||||
environment:
|
environment:
|
||||||
|
- PRELOAD=true
|
||||||
- CLI_ARGS=
|
- CLI_ARGS=
|
||||||
|
|
||||||
|
|
||||||
comfy-cpu:
|
|
||||||
<<: *comfy
|
|
||||||
profiles: ["comfy-cpu"]
|
|
||||||
deploy: {}
|
|
||||||
environment:
|
|
||||||
- CLI_ARGS=--cpu
|
|
||||||
|
|
|
||||||
|
|
@ -1,66 +1,90 @@
|
||||||
|
# syntax=docker/dockerfile:1
|
||||||
|
|
||||||
FROM alpine/git:2.36.2 as download
|
FROM alpine/git:2.36.2 as download
|
||||||
|
|
||||||
COPY clone.sh /clone.sh
|
SHELL ["/bin/sh", "-ceuxo", "pipefail"]
|
||||||
|
|
||||||
RUN . /clone.sh stable-diffusion-webui-assets https://github.com/AUTOMATIC1111/stable-diffusion-webui-assets.git 6f7db241d2f8ba7457bac5ca9753331f0c266917
|
RUN <<EOF
|
||||||
|
cat <<'EOE' > /clone.sh
|
||||||
|
mkdir -p repositories/"$1" && cd repositories/"$1" && git init && git remote add origin "$2" && git fetch origin "$3" --depth=1 && git reset --hard "$3" && rm -rf .git
|
||||||
|
EOE
|
||||||
|
EOF
|
||||||
|
|
||||||
RUN . /clone.sh stable-diffusion-stability-ai https://github.com/Stability-AI/stablediffusion.git cf1d67a6fd5ea1aa600c4df58e5b47da45f6bdbf \
|
RUN . /clone.sh taming-transformers https://github.com/CompVis/taming-transformers.git 24268930bf1dce879235a7fddd0b2355b84d7ea6 \
|
||||||
|
&& rm -rf data assets **/*.ipynb
|
||||||
|
|
||||||
|
RUN . /clone.sh stable-diffusion https://github.com/CompVis/stable-diffusion.git 69ae4b35e0a0f6ee1af8bb9a5d0016ccb27e36dc \
|
||||||
&& rm -rf assets data/**/*.png data/**/*.jpg data/**/*.gif
|
&& rm -rf assets data/**/*.png data/**/*.jpg data/**/*.gif
|
||||||
|
|
||||||
|
RUN . /clone.sh CodeFormer https://github.com/sczhou/CodeFormer.git c5b4593074ba6214284d6acd5f1719b6c5d739af \
|
||||||
|
&& rm -rf assets inputs
|
||||||
|
|
||||||
RUN . /clone.sh BLIP https://github.com/salesforce/BLIP.git 48211a1594f1321b00f14c9f7a5b4813144b2fb9
|
RUN . /clone.sh BLIP https://github.com/salesforce/BLIP.git 48211a1594f1321b00f14c9f7a5b4813144b2fb9
|
||||||
RUN . /clone.sh k-diffusion https://github.com/crowsonkb/k-diffusion.git ab527a9a6d347f364e3d185ba6d714e22d80cb3c
|
RUN . /clone.sh k-diffusion https://github.com/crowsonkb/k-diffusion.git f4e99857772fc3a126ba886aadf795a332774878
|
||||||
RUN . /clone.sh clip-interrogator https://github.com/pharmapsychotic/clip-interrogator 2cf03aaf6e704197fd0dae7c7f96aa59cf1b11c9
|
RUN . /clone.sh clip-interrogator https://github.com/pharmapsychotic/clip-interrogator 2486589f24165c8e3b303f84e9dbbea318df83e8
|
||||||
RUN . /clone.sh generative-models https://github.com/Stability-AI/generative-models 45c443b316737a4ab6e40413d7794a7f5657c19f
|
|
||||||
RUN . /clone.sh stable-diffusion-webui-assets https://github.com/AUTOMATIC1111/stable-diffusion-webui-assets 6f7db241d2f8ba7457bac5ca9753331f0c266917
|
|
||||||
|
|
||||||
|
|
||||||
FROM pytorch/pytorch:2.3.0-cuda12.1-cudnn8-runtime
|
FROM alpine:3 as xformers
|
||||||
|
RUN apk add aria2
|
||||||
|
RUN aria2c --dir / --out wheel.whl 'https://github.com/AbdBarho/stable-diffusion-webui-docker/releases/download/2.1.0/xformers-0.0.14.dev0-cp310-cp310-linux_x86_64.whl'
|
||||||
|
|
||||||
ENV DEBIAN_FRONTEND=noninteractive PIP_PREFER_BINARY=1
|
FROM python:3.10-slim
|
||||||
|
|
||||||
RUN --mount=type=cache,target=/var/cache/apt \
|
SHELL ["/bin/bash", "-ceuxo", "pipefail"]
|
||||||
apt-get update && \
|
|
||||||
# we need those
|
ENV DEBIAN_FRONTEND=noninteractive PIP_PREFER_BINARY=1 PIP_NO_CACHE_DIR=1
|
||||||
apt-get install -y fonts-dejavu-core rsync git jq moreutils aria2 \
|
|
||||||
# extensions needs those
|
RUN pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 --extra-index-url https://download.pytorch.org/whl/cu113
|
||||||
ffmpeg libglfw3-dev libgles2-mesa-dev pkg-config libcairo2 libcairo2-dev build-essential
|
|
||||||
|
RUN apt-get update && apt install fonts-dejavu-core rsync git jq moreutils -y && apt-get clean
|
||||||
|
|
||||||
|
|
||||||
WORKDIR /
|
RUN <<EOF
|
||||||
RUN --mount=type=cache,target=/root/.cache/pip \
|
git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
|
||||||
git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git && \
|
cd stable-diffusion-webui
|
||||||
cd stable-diffusion-webui && \
|
git reset --hard d885a4a57b72152745ca76192ef1bdda29e6461d
|
||||||
git reset --hard v1.9.4 && \
|
pip install -r requirements_versions.txt
|
||||||
pip install -r requirements_versions.txt
|
EOF
|
||||||
|
|
||||||
|
|
||||||
|
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 && rm xformers-0.0.14.dev0-cp310-cp310-linux_x86_64.whl
|
||||||
|
|
||||||
|
|
||||||
ENV ROOT=/stable-diffusion-webui
|
ENV ROOT=/stable-diffusion-webui
|
||||||
|
|
||||||
COPY --from=download /repositories/ ${ROOT}/repositories/
|
|
||||||
RUN mkdir ${ROOT}/interrogate && cp ${ROOT}/repositories/clip-interrogator/clip_interrogator/data/* ${ROOT}/interrogate
|
|
||||||
|
|
||||||
RUN --mount=type=cache,target=/root/.cache/pip \
|
COPY --from=download /git/ ${ROOT}
|
||||||
pip install pyngrok xformers==0.0.26.post1 \
|
RUN mkdir ${ROOT}/interrogate && cp ${ROOT}/repositories/clip-interrogator/data/* ${ROOT}/interrogate
|
||||||
|
RUN pip install --prefer-binary --no-cache-dir -r ${ROOT}/repositories/CodeFormer/requirements.txt
|
||||||
|
|
||||||
|
|
||||||
|
# Note: don't update the sha of previous versions because the install will take forever
|
||||||
|
# instead, update the repo state in a later step
|
||||||
|
|
||||||
|
ARG SHA=35c45df28b303a05d56a13cb56d4046f08cf8c25
|
||||||
|
RUN <<EOF
|
||||||
|
cd stable-diffusion-webui
|
||||||
|
git fetch
|
||||||
|
git reset --hard ${SHA}
|
||||||
|
pip install -r requirements_versions.txt
|
||||||
|
EOF
|
||||||
|
|
||||||
|
RUN pip install opencv-python-headless \
|
||||||
git+https://github.com/TencentARC/GFPGAN.git@8d2447a2d918f8eba5a4a01463fd48e45126a379 \
|
git+https://github.com/TencentARC/GFPGAN.git@8d2447a2d918f8eba5a4a01463fd48e45126a379 \
|
||||||
git+https://github.com/openai/CLIP.git@d50d76daa670286dd6cacf3bcd80b5e4823fc8e1 \
|
git+https://github.com/openai/CLIP.git@d50d76daa670286dd6cacf3bcd80b5e4823fc8e1 \
|
||||||
git+https://github.com/mlfoundations/open_clip.git@v2.20.0
|
pyngrok \
|
||||||
|
gradio==3.4.1
|
||||||
|
# gradio because #173 & https://github.com/AUTOMATIC1111/stable-diffusion-webui/issues/2858
|
||||||
|
|
||||||
# there seems to be a memory leak (or maybe just memory not being freed fast enough) that is fixed by this version of malloc
|
|
||||||
# maybe move this up to the dependencies list.
|
|
||||||
RUN apt-get -y install libgoogle-perftools-dev && apt-get clean
|
|
||||||
ENV LD_PRELOAD=libtcmalloc.so
|
|
||||||
|
|
||||||
COPY . /docker
|
COPY . /docker
|
||||||
|
|
||||||
RUN \
|
RUN chmod +x /docker/entrypoint.sh && python3 /docker/info.py ${ROOT}/modules/ui.py
|
||||||
# mv ${ROOT}/style.css ${ROOT}/user.css && \
|
|
||||||
# one of the ugliest hacks I ever wrote \
|
|
||||||
sed -i 's/in_app_dir = .*/in_app_dir = True/g' /opt/conda/lib/python3.10/site-packages/gradio/routes.py && \
|
|
||||||
git config --global --add safe.directory '*'
|
|
||||||
|
|
||||||
WORKDIR ${ROOT}
|
WORKDIR ${ROOT}/repositories/stable-diffusion
|
||||||
ENV NVIDIA_VISIBLE_DEVICES=all
|
|
||||||
ENV CLI_ARGS=""
|
ENV CLI_ARGS=""
|
||||||
EXPOSE 7860
|
EXPOSE 7860
|
||||||
ENTRYPOINT ["/docker/entrypoint.sh"]
|
ENTRYPOINT ["/docker/entrypoint.sh"]
|
||||||
CMD python -u webui.py --listen --port 7860 ${CLI_ARGS}
|
# run, -u to not buffer stdout / stderr
|
||||||
|
CMD python3 -u ../../webui.py --listen --port 7860 --ckpt-dir ${ROOT}/models/Stable-diffusion --theme dark ${CLI_ARGS}
|
||||||
|
|
|
||||||
|
|
@ -1,11 +0,0 @@
|
||||||
#!/bin/bash
|
|
||||||
|
|
||||||
set -Eeuox pipefail
|
|
||||||
|
|
||||||
mkdir -p /repositories/"$1"
|
|
||||||
cd /repositories/"$1"
|
|
||||||
git init
|
|
||||||
git remote add origin "$2"
|
|
||||||
git fetch origin "$3" --depth=1
|
|
||||||
git reset --hard "$3"
|
|
||||||
rm -rf .git
|
|
||||||
10
services/AUTOMATIC1111/config.json
Normal file
10
services/AUTOMATIC1111/config.json
Normal file
|
|
@ -0,0 +1,10 @@
|
||||||
|
{
|
||||||
|
"outdir_samples": "/output",
|
||||||
|
"outdir_txt2img_samples": "/output/txt2img-images",
|
||||||
|
"outdir_img2img_samples": "/output/img2img-images",
|
||||||
|
"outdir_extras_samples": "/output/extras-images",
|
||||||
|
"outdir_txt2img_grids": "/output/txt2img-grids",
|
||||||
|
"outdir_img2img_grids": "/output/img2img-grids",
|
||||||
|
"outdir_save": "/output/saved",
|
||||||
|
"font": "DejaVuSans.ttf"
|
||||||
|
}
|
||||||
|
|
@ -1,78 +0,0 @@
|
||||||
#!/usr/bin/env python3
|
|
||||||
|
|
||||||
"""Checks and sets default values for config.json before starting the container."""
|
|
||||||
|
|
||||||
import json
|
|
||||||
import re
|
|
||||||
import os.path
|
|
||||||
import sys
|
|
||||||
|
|
||||||
DEFAULT_FILEPATH = '/data/config/auto/config.json'
|
|
||||||
|
|
||||||
DEFAULT_OUTDIRS = {
|
|
||||||
"outdir_samples": "",
|
|
||||||
"outdir_txt2img_samples": "/output/txt2img",
|
|
||||||
"outdir_img2img_samples": "/output/img2img",
|
|
||||||
"outdir_extras_samples": "/output/extras",
|
|
||||||
"outdir_grids": "",
|
|
||||||
"outdir_txt2img_grids": "/output/txt2img-grids",
|
|
||||||
"outdir_img2img_grids": "/output/img2img-grids",
|
|
||||||
"outdir_save": "/output/saved",
|
|
||||||
"outdir_init_images": "/output/init-images",
|
|
||||||
}
|
|
||||||
RE_VALID_OUTDIR = re.compile(r"(^/output(/\.?[\w\-\_]+)+/?$)|(^\s?$)")
|
|
||||||
|
|
||||||
DEFAULT_OTHER = {
|
|
||||||
"font": "DejaVuSans.ttf",
|
|
||||||
}
|
|
||||||
|
|
||||||
def dict_to_json_file(target_file: str, data: dict):
|
|
||||||
"""Write dictionary to specified json file"""
|
|
||||||
|
|
||||||
with open(target_file, 'w') as f:
|
|
||||||
json.dump(data, f)
|
|
||||||
|
|
||||||
def json_file_to_dict(config_file: str) -> dict|None:
|
|
||||||
"""Load json file into a dictionary. Return None if file does not exist."""
|
|
||||||
|
|
||||||
if os.path.isfile(config_file):
|
|
||||||
with open(config_file, 'r') as f:
|
|
||||||
return json.load(f)
|
|
||||||
else:
|
|
||||||
return None
|
|
||||||
|
|
||||||
def replace_if_invalid(value: str, replacement: str, pattern: str|re.Pattern[str]) -> str:
|
|
||||||
"""Returns original value if valid, fallback value if invalid"""
|
|
||||||
|
|
||||||
if re.match(pattern, value):
|
|
||||||
return value
|
|
||||||
else:
|
|
||||||
return replacement
|
|
||||||
|
|
||||||
def check_and_replace_config(config_file: str, target_file: str = None):
|
|
||||||
"""Checks given file for invalid values. Replaces those with fallback values (default: overwrites file)."""
|
|
||||||
|
|
||||||
# Get current user config, or empty if file does not exists
|
|
||||||
data = json_file_to_dict(config_file) or {}
|
|
||||||
|
|
||||||
# Check and fix output directories
|
|
||||||
for k, def_val in DEFAULT_OUTDIRS.items():
|
|
||||||
if k not in data:
|
|
||||||
data[k] = def_val
|
|
||||||
else:
|
|
||||||
data[k] = replace_if_invalid(value=data[k], replacement=def_val, pattern=RE_VALID_OUTDIR)
|
|
||||||
|
|
||||||
# Check and fix other default settings
|
|
||||||
for k, def_val in DEFAULT_OTHER.items():
|
|
||||||
if k not in data:
|
|
||||||
data[k] = def_val
|
|
||||||
|
|
||||||
# Write results to file
|
|
||||||
dict_to_json_file(target_file or config_file, data)
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
|
||||||
if len(sys.argv) > 1:
|
|
||||||
check_and_replace_config(*sys.argv[1:])
|
|
||||||
else:
|
|
||||||
check_and_replace_config(DEFAULT_FILEPATH)
|
|
||||||
|
|
||||||
|
|
@ -4,42 +4,36 @@ set -Eeuo pipefail
|
||||||
|
|
||||||
# TODO: move all mkdir -p ?
|
# TODO: move all mkdir -p ?
|
||||||
mkdir -p /data/config/auto/scripts/
|
mkdir -p /data/config/auto/scripts/
|
||||||
# mount scripts individually
|
cp -n /docker/config.json /data/config/auto/config.json
|
||||||
|
jq '. * input' /data/config/auto/config.json /docker/config.json | sponge /data/config/auto/config.json
|
||||||
echo $ROOT
|
|
||||||
ls -lha $ROOT
|
|
||||||
|
|
||||||
find "${ROOT}/scripts/" -maxdepth 1 -type l -delete
|
|
||||||
cp -vrfTs /data/config/auto/scripts/ "${ROOT}/scripts/"
|
|
||||||
|
|
||||||
# Set up config file
|
|
||||||
python /docker/config.py /data/config/auto/config.json
|
|
||||||
|
|
||||||
if [ ! -f /data/config/auto/ui-config.json ]; then
|
if [ ! -f /data/config/auto/ui-config.json ]; then
|
||||||
echo '{}' >/data/config/auto/ui-config.json
|
echo '{}' >/data/config/auto/ui-config.json
|
||||||
fi
|
fi
|
||||||
|
|
||||||
if [ ! -f /data/config/auto/styles.csv ]; then
|
# copy scripts, we cannot just mount the directory because it will override the already provided scripts in the repo
|
||||||
touch /data/config/auto/styles.csv
|
cp -rfT /data/config/auto/scripts/ "${ROOT}/scripts"
|
||||||
fi
|
|
||||||
|
|
||||||
# copy models from original models folder
|
|
||||||
mkdir -p /data/models/VAE-approx/ /data/models/karlo/
|
|
||||||
|
|
||||||
rsync -a --info=NAME ${ROOT}/models/VAE-approx/ /data/models/VAE-approx/
|
|
||||||
rsync -a --info=NAME ${ROOT}/models/karlo/ /data/models/karlo/
|
|
||||||
|
|
||||||
declare -A MOUNTS
|
declare -A MOUNTS
|
||||||
|
|
||||||
MOUNTS["/root/.cache"]="/data/.cache"
|
MOUNTS["/root/.cache"]="/data/.cache"
|
||||||
MOUNTS["${ROOT}/models"]="/data/models"
|
|
||||||
|
# main
|
||||||
|
MOUNTS["${ROOT}/models/Stable-diffusion"]="/data/StableDiffusion"
|
||||||
|
MOUNTS["${ROOT}/models/Codeformer"]="/data/Codeformer"
|
||||||
|
MOUNTS["${ROOT}/models/GFPGAN"]="/data/GFPGAN"
|
||||||
|
MOUNTS["${ROOT}/models/ESRGAN"]="/data/ESRGAN"
|
||||||
|
MOUNTS["${ROOT}/models/BSRGAN"]="/data/BSRGAN"
|
||||||
|
MOUNTS["${ROOT}/models/RealESRGAN"]="/data/RealESRGAN"
|
||||||
|
MOUNTS["${ROOT}/models/SwinIR"]="/data/SwinIR"
|
||||||
|
MOUNTS["${ROOT}/models/ScuNET"]="/data/ScuNET"
|
||||||
|
MOUNTS["${ROOT}/models/LDSR"]="/data/LDSR"
|
||||||
|
MOUNTS["${ROOT}/models/hypernetworks"]="/data/Hypernetworks"
|
||||||
|
|
||||||
MOUNTS["${ROOT}/embeddings"]="/data/embeddings"
|
MOUNTS["${ROOT}/embeddings"]="/data/embeddings"
|
||||||
MOUNTS["${ROOT}/config.json"]="/data/config/auto/config.json"
|
MOUNTS["${ROOT}/config.json"]="/data/config/auto/config.json"
|
||||||
MOUNTS["${ROOT}/ui-config.json"]="/data/config/auto/ui-config.json"
|
MOUNTS["${ROOT}/ui-config.json"]="/data/config/auto/ui-config.json"
|
||||||
MOUNTS["${ROOT}/styles.csv"]="/data/config/auto/styles.csv"
|
|
||||||
MOUNTS["${ROOT}/extensions"]="/data/config/auto/extensions"
|
MOUNTS["${ROOT}/extensions"]="/data/config/auto/extensions"
|
||||||
MOUNTS["${ROOT}/config_states"]="/data/config/auto/config_states"
|
|
||||||
|
|
||||||
# extra hacks
|
# extra hacks
|
||||||
MOUNTS["${ROOT}/repositories/CodeFormer/weights/facelib"]="/data/.cache"
|
MOUNTS["${ROOT}/repositories/CodeFormer/weights/facelib"]="/data/.cache"
|
||||||
|
|
@ -56,30 +50,13 @@ for to_path in "${!MOUNTS[@]}"; do
|
||||||
echo Mounted $(basename "${from_path}")
|
echo Mounted $(basename "${from_path}")
|
||||||
done
|
done
|
||||||
|
|
||||||
echo "Installing extension dependencies (if any)"
|
mkdir -p /output/saved /output/txt2img-images/ /output/img2img-images /output/extras-images/ /output/grids/ /output/txt2img-grids/ /output/img2img-grids/
|
||||||
|
|
||||||
# because we build our container as root:
|
|
||||||
chown -R root ~/.cache/
|
|
||||||
chmod 766 ~/.cache/
|
|
||||||
|
|
||||||
shopt -s nullglob
|
|
||||||
# For install.py, please refer to https://github.com/AUTOMATIC1111/stable-diffusion-webui/wiki/Developing-extensions#installpy
|
|
||||||
list=(./extensions/*/install.py)
|
|
||||||
for installscript in "${list[@]}"; do
|
|
||||||
EXTNAME=$(echo $installscript | cut -d '/' -f 3)
|
|
||||||
# Skip installing dependencies if extension is disabled in config
|
|
||||||
if $(jq -e ".disabled_extensions|any(. == \"$EXTNAME\")" config.json); then
|
|
||||||
echo "Skipping disabled extension ($EXTNAME)"
|
|
||||||
continue
|
|
||||||
fi
|
|
||||||
PYTHONPATH=${ROOT} python "$installscript"
|
|
||||||
done
|
|
||||||
|
|
||||||
if [ -f "/data/config/auto/startup.sh" ]; then
|
if [ -f "/data/config/auto/startup.sh" ]; then
|
||||||
pushd ${ROOT}
|
pushd ${ROOT}
|
||||||
echo "Running startup script"
|
|
||||||
. /data/config/auto/startup.sh
|
. /data/config/auto/startup.sh
|
||||||
popd
|
popd
|
||||||
fi
|
fi
|
||||||
|
|
||||||
|
|
||||||
exec "$@"
|
exec "$@"
|
||||||
|
|
|
||||||
14
services/AUTOMATIC1111/info.py
Normal file
14
services/AUTOMATIC1111/info.py
Normal file
|
|
@ -0,0 +1,14 @@
|
||||||
|
import sys
|
||||||
|
from pathlib import Path
|
||||||
|
|
||||||
|
file = Path(sys.argv[1])
|
||||||
|
file.write_text(
|
||||||
|
file.read_text()\
|
||||||
|
.replace(' return demo', """
|
||||||
|
with demo:
|
||||||
|
gr.Markdown(
|
||||||
|
'Created by [AUTOMATIC1111 / stable-diffusion-webui-docker](https://github.com/AbdBarho/stable-diffusion-webui-docker/)'
|
||||||
|
)
|
||||||
|
return demo
|
||||||
|
""", 1)
|
||||||
|
)
|
||||||
|
|
@ -1,22 +0,0 @@
|
||||||
FROM pytorch/pytorch:2.3.0-cuda12.1-cudnn8-runtime
|
|
||||||
|
|
||||||
ENV DEBIAN_FRONTEND=noninteractive PIP_PREFER_BINARY=1
|
|
||||||
|
|
||||||
RUN apt-get update && apt-get install -y git && apt-get clean
|
|
||||||
|
|
||||||
ENV ROOT=/stable-diffusion
|
|
||||||
RUN --mount=type=cache,target=/root/.cache/pip \
|
|
||||||
git clone https://github.com/comfyanonymous/ComfyUI.git ${ROOT} && \
|
|
||||||
cd ${ROOT} && \
|
|
||||||
git checkout master && \
|
|
||||||
git reset --hard 276f8fce9f5a80b500947fb5745a4dde9e84622d && \
|
|
||||||
pip install -r requirements.txt
|
|
||||||
|
|
||||||
WORKDIR ${ROOT}
|
|
||||||
COPY . /docker/
|
|
||||||
RUN chmod u+x /docker/entrypoint.sh && cp /docker/extra_model_paths.yaml ${ROOT}
|
|
||||||
|
|
||||||
ENV NVIDIA_VISIBLE_DEVICES=all PYTHONPATH="${PYTHONPATH}:${PWD}" CLI_ARGS=""
|
|
||||||
EXPOSE 7860
|
|
||||||
ENTRYPOINT ["/docker/entrypoint.sh"]
|
|
||||||
CMD python -u main.py --listen --port 7860 ${CLI_ARGS}
|
|
||||||
|
|
@ -1,31 +0,0 @@
|
||||||
#!/bin/bash
|
|
||||||
|
|
||||||
set -Eeuo pipefail
|
|
||||||
|
|
||||||
mkdir -vp /data/config/comfy/custom_nodes
|
|
||||||
|
|
||||||
declare -A MOUNTS
|
|
||||||
|
|
||||||
MOUNTS["/root/.cache"]="/data/.cache"
|
|
||||||
MOUNTS["${ROOT}/input"]="/data/config/comfy/input"
|
|
||||||
MOUNTS["${ROOT}/output"]="/output/comfy"
|
|
||||||
|
|
||||||
for to_path in "${!MOUNTS[@]}"; do
|
|
||||||
set -Eeuo pipefail
|
|
||||||
from_path="${MOUNTS[${to_path}]}"
|
|
||||||
rm -rf "${to_path}"
|
|
||||||
if [ ! -f "$from_path" ]; then
|
|
||||||
mkdir -vp "$from_path"
|
|
||||||
fi
|
|
||||||
mkdir -vp "$(dirname "${to_path}")"
|
|
||||||
ln -sT "${from_path}" "${to_path}"
|
|
||||||
echo Mounted $(basename "${from_path}")
|
|
||||||
done
|
|
||||||
|
|
||||||
if [ -f "/data/config/comfy/startup.sh" ]; then
|
|
||||||
pushd ${ROOT}
|
|
||||||
. /data/config/comfy/startup.sh
|
|
||||||
popd
|
|
||||||
fi
|
|
||||||
|
|
||||||
exec "$@"
|
|
||||||
|
|
@ -1,25 +0,0 @@
|
||||||
a111:
|
|
||||||
base_path: /data
|
|
||||||
|
|
||||||
checkpoints: models/Stable-diffusion
|
|
||||||
configs: models/Stable-diffusion
|
|
||||||
vae: models/VAE
|
|
||||||
loras: models/Lora
|
|
||||||
upscale_models: |
|
|
||||||
models/RealESRGAN
|
|
||||||
models/ESRGAN
|
|
||||||
models/SwinIR
|
|
||||||
models/GFPGAN
|
|
||||||
hypernetworks: models/hypernetworks
|
|
||||||
controlnet: models/ControlNet
|
|
||||||
gligen: models/GLIGEN
|
|
||||||
clip: models/CLIPEncoder
|
|
||||||
embeddings: embeddings
|
|
||||||
|
|
||||||
custom_nodes: config/comfy/custom_nodes
|
|
||||||
|
|
||||||
# TODO: I am unsure about these, need more testing
|
|
||||||
# style_models: config/comfy/style_models
|
|
||||||
# t2i_adapter: config/comfy/t2i_adapter
|
|
||||||
# clip_vision: config/comfy/clip_vision
|
|
||||||
# diffusers: config/comfy/diffusers
|
|
||||||
|
|
@ -1,6 +1,6 @@
|
||||||
FROM bash:alpine3.19
|
FROM bash:alpine3.15
|
||||||
|
|
||||||
RUN apk update && apk add parallel aria2
|
RUN apk add parallel aria2
|
||||||
COPY . /docker
|
COPY . /docker
|
||||||
RUN chmod +x /docker/download.sh
|
RUN chmod +x /docker/download.sh
|
||||||
ENTRYPOINT ["/docker/download.sh"]
|
ENTRYPOINT ["/docker/download.sh"]
|
||||||
|
|
|
||||||
|
|
@ -1,8 +1,6 @@
|
||||||
cc6cb27103417325ff94f52b7a5d2dde45a7515b25c255d8e396c90014281516 /data/models/Stable-diffusion/v1-5-pruned-emaonly.ckpt
|
fe4efff1e174c627256e44ec2991ba279b3816e364b49f9be2abc0b3ff3f8556 /data/StableDiffusion/model.ckpt
|
||||||
c6bbc15e3224e6973459ba78de4998b80b50112b0ae5b5c67113d56b4e366b19 /data/models/Stable-diffusion/sd-v1-5-inpainting.ckpt
|
e2cd4703ab14f4d01fd1383a8a8b266f9a5833dacee8e6a79d3bf21a1b6be5ad /data/GFPGAN/GFPGANv1.4.pth
|
||||||
c6a580b13a5bc05a5e16e4dbb80608ff2ec251a162311590c1f34c013d7f3dab /data/models/VAE/vae-ft-mse-840000-ema-pruned.ckpt
|
4fa0d38905f75ac06eb49a7951b426670021be3018265fd191d2125df9d682f1 /data/RealESRGAN/RealESRGAN_x4plus.pth
|
||||||
e2cd4703ab14f4d01fd1383a8a8b266f9a5833dacee8e6a79d3bf21a1b6be5ad /data/models/GFPGAN/GFPGANv1.4.pth
|
f872d837d3c90ed2e05227bed711af5671a6fd1c9f7d7e91c911a61f155e99da /data/RealESRGAN/RealESRGAN_x4plus_anime_6B.pth
|
||||||
4fa0d38905f75ac06eb49a7951b426670021be3018265fd191d2125df9d682f1 /data/models/RealESRGAN/RealESRGAN_x4plus.pth
|
c209caecac2f97b4bb8f4d726b70ac2ac9b35904b7fc99801e1f5e61f9210c13 /data/LDSR/model.ckpt
|
||||||
f872d837d3c90ed2e05227bed711af5671a6fd1c9f7d7e91c911a61f155e99da /data/models/RealESRGAN/RealESRGAN_x4plus_anime_6B.pth
|
9d6ad53c5dafeb07200fb712db14b813b527edd262bc80ea136777bdb41be2ba /data/LDSR/project.yaml
|
||||||
c209caecac2f97b4bb8f4d726b70ac2ac9b35904b7fc99801e1f5e61f9210c13 /data/models/LDSR/model.ckpt
|
|
||||||
9d6ad53c5dafeb07200fb712db14b813b527edd262bc80ea136777bdb41be2ba /data/models/LDSR/project.yaml
|
|
||||||
|
|
|
||||||
|
|
@ -2,30 +2,28 @@
|
||||||
|
|
||||||
set -Eeuo pipefail
|
set -Eeuo pipefail
|
||||||
|
|
||||||
# TODO: maybe just use the .gitignore file to create all of these
|
mkdir -p /data/.cache /data/StableDiffusion /data/Codeformer /data/GFPGAN /data/ESRGAN /data/BSRGAN /data/RealESRGAN /data/SwinIR /data/LDSR /data/ScuNET /data/embeddings
|
||||||
mkdir -vp /data/.cache \
|
|
||||||
/data/embeddings \
|
|
||||||
/data/config/ \
|
|
||||||
/data/models/ \
|
|
||||||
/data/models/Stable-diffusion \
|
|
||||||
/data/models/GFPGAN \
|
|
||||||
/data/models/RealESRGAN \
|
|
||||||
/data/models/LDSR \
|
|
||||||
/data/models/VAE
|
|
||||||
|
|
||||||
echo "Downloading, this might take a while..."
|
echo "Downloading, this might take a while..."
|
||||||
|
|
||||||
aria2c -x 10 --disable-ipv6 --input-file /docker/links.txt --dir /data/models --continue
|
aria2c --disable-ipv6 --input-file /docker/links.txt --dir /data --continue
|
||||||
|
|
||||||
echo "Checking SHAs..."
|
echo "Checking SHAs..."
|
||||||
|
|
||||||
parallel --will-cite -a /docker/checksums.sha256 "echo -n {} | sha256sum -c"
|
parallel --will-cite -a /docker/checksums.sha256 "echo -n {} | sha256sum -c"
|
||||||
|
|
||||||
|
# aria2c already does hash check
|
||||||
|
# cc6cb27103417325ff94f52b7a5d2dde45a7515b25c255d8e396c90014281516 /data/StableDiffusion/v1-5-pruned-emaonly.ckpt
|
||||||
cat <<EOF
|
cat <<EOF
|
||||||
By using this software, you agree to the following licenses:
|
By using this software, you agree to the following licenses:
|
||||||
https://github.com/AbdBarho/stable-diffusion-webui-docker/blob/master/LICENSE
|
|
||||||
https://github.com/CompVis/stable-diffusion/blob/main/LICENSE
|
https://github.com/CompVis/stable-diffusion/blob/main/LICENSE
|
||||||
https://github.com/AUTOMATIC1111/stable-diffusion-webui/blob/master/LICENSE.txt
|
https://github.com/AbdBarho/stable-diffusion-webui-docker/blob/master/LICENSE
|
||||||
|
https://github.com/sd-webui/stable-diffusion-webui/blob/master/LICENSE
|
||||||
https://github.com/invoke-ai/InvokeAI/blob/main/LICENSE
|
https://github.com/invoke-ai/InvokeAI/blob/main/LICENSE
|
||||||
And licenses of all UIs, third party libraries, and extensions.
|
https://github.com/cszn/BSRGAN/blob/main/LICENSE
|
||||||
|
https://github.com/sczhou/CodeFormer/blob/master/LICENSE
|
||||||
|
https://github.com/TencentARC/GFPGAN/blob/master/LICENSE
|
||||||
|
https://github.com/xinntao/Real-ESRGAN/blob/master/LICENSE
|
||||||
|
https://github.com/xinntao/ESRGAN/blob/master/LICENSE
|
||||||
|
https://github.com/cszn/SCUNet/blob/main/LICENSE
|
||||||
EOF
|
EOF
|
||||||
|
|
|
||||||
|
|
@ -1,9 +1,13 @@
|
||||||
https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/v1-5-pruned-emaonly.ckpt
|
# 'check-integrity=true' is the only way aria2c won't fail if the file already exists
|
||||||
out=Stable-diffusion/v1-5-pruned-emaonly.ckpt
|
magnet:?xt=urn:btih:2daef5b5f63a16a9af9169a529b1a773fc452637&dn=v1-5-pruned-emaonly.ckpt&tr=udp%3a%2f%2ftracker.opentrackr.org%3a1337%2fannounce&tr=udp%3a%2f%2f9.rarbg.com%3a2810%2fannounce&tr=udp%3a%2f%2ftracker.openbittorrent.com%3a6969%2fannounce&tr=udp%3a%2f%2fopentracker.i2p.rocks%3a6969%2fannounce&tr=https%3a%2f%2fopentracker.i2p.rocks%3a443%2fannounce&tr=http%3a%2f%2ftracker.openbittorrent.com%3a80%2fannounce&tr=udp%3a%2f%2ftracker.torrent.eu.org%3a451%2fannounce&tr=udp%3a%2f%2fopen.stealth.si%3a80%2fannounce&tr=udp%3a%2f%2fvibe.sleepyinternetfun.xyz%3a1738%2fannounce&tr=udp%3a%2f%2ftracker2.dler.org%3a80%2fannounce&tr=udp%3a%2f%2ftracker1.bt.moack.co.kr%3a80%2fannounce&tr=udp%3a%2f%2ftracker.zemoj.com%3a6969%2fannounce&tr=udp%3a%2f%2ftracker.tiny-vps.com%3a6969%2fannounce&tr=udp%3a%2f%2ftracker.theoks.net%3a6969%2fannounce&tr=udp%3a%2f%2ftracker.publictracker.xyz%3a6969%2fannounce&tr=udp%3a%2f%2ftracker.monitorit4.me%3a6969%2fannounce&tr=udp%3a%2f%2ftracker.moeking.me%3a6969%2fannounce&tr=udp%3a%2f%2ftracker.lelux.fi%3a6969%2fannounce&tr=udp%3a%2f%2ftracker.dler.org%3a6969%2fannounce&tr=udp%3a%2f%2ftracker.army%3a6969%2fannounce
|
||||||
https://huggingface.co/stabilityai/sd-vae-ft-mse-original/resolve/main/vae-ft-mse-840000-ema-pruned.ckpt
|
select-file=1
|
||||||
out=VAE/vae-ft-mse-840000-ema-pruned.ckpt
|
index-out=1=StableDiffusion/v1-5-pruned-emaonly.ckpt
|
||||||
https://huggingface.co/runwayml/stable-diffusion-inpainting/resolve/main/sd-v1-5-inpainting.ckpt
|
follow-torrent=mem
|
||||||
out=Stable-diffusion/sd-v1-5-inpainting.ckpt
|
follow-metalink=mem
|
||||||
|
seed-time=0
|
||||||
|
check-integrity=true
|
||||||
|
https://drive.yerf.org/wl/?id=EBfTrmcCCUAGaQBXVIj5lJmEhjoP1tgl&mode=grid&download=1
|
||||||
|
out=StableDiffusion/model.ckpt
|
||||||
https://github.com/TencentARC/GFPGAN/releases/download/v1.3.4/GFPGANv1.4.pth
|
https://github.com/TencentARC/GFPGAN/releases/download/v1.3.4/GFPGANv1.4.pth
|
||||||
out=GFPGAN/GFPGANv1.4.pth
|
out=GFPGAN/GFPGANv1.4.pth
|
||||||
https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.0/RealESRGAN_x4plus.pth
|
https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.0/RealESRGAN_x4plus.pth
|
||||||
|
|
|
||||||
51
services/hlky/Dockerfile
Normal file
51
services/hlky/Dockerfile
Normal file
|
|
@ -0,0 +1,51 @@
|
||||||
|
# syntax=docker/dockerfile:1
|
||||||
|
|
||||||
|
FROM continuumio/miniconda3:4.12.0
|
||||||
|
|
||||||
|
SHELL ["/bin/bash", "-ceuxo", "pipefail"]
|
||||||
|
|
||||||
|
ENV DEBIAN_FRONTEND=noninteractive
|
||||||
|
|
||||||
|
RUN conda install python=3.8.5 && conda clean -a -y
|
||||||
|
RUN conda install pytorch==1.11.0 torchvision==0.12.0 cudatoolkit=11.3 -c pytorch && conda clean -a -y
|
||||||
|
|
||||||
|
RUN apt-get update && apt install fonts-dejavu-core rsync gcc -y && apt-get clean
|
||||||
|
|
||||||
|
|
||||||
|
ENV PIP_PREFER_BINARY=1 PIP_NO_CACHE_DIR=1
|
||||||
|
|
||||||
|
RUN <<EOF
|
||||||
|
git config --global http.postBuffer 1048576000
|
||||||
|
git clone https://github.com/Sygil-Dev/sygil-webui.git stable-diffusion
|
||||||
|
cd stable-diffusion
|
||||||
|
git reset --hard 091520bed06f913c9f432f9f47ccbe22b46068d7
|
||||||
|
conda env update --file environment.yaml -n base
|
||||||
|
conda clean -a -y
|
||||||
|
EOF
|
||||||
|
|
||||||
|
RUN apt-get update && apt install libsndfile1 ffmpeg -y && apt-get clean
|
||||||
|
|
||||||
|
ARG BRANCH=master SHA=091520bed06f913c9f432f9f47ccbe22b46068d7
|
||||||
|
RUN <<EOF
|
||||||
|
cd stable-diffusion
|
||||||
|
git fetch
|
||||||
|
git checkout ${BRANCH}
|
||||||
|
git reset --hard ${SHA}
|
||||||
|
conda env update --file environment.yaml -n base
|
||||||
|
conda clean -a -y
|
||||||
|
EOF
|
||||||
|
|
||||||
|
# add info
|
||||||
|
COPY . /docker/
|
||||||
|
RUN <<EOF
|
||||||
|
python /docker/info.py /stable-diffusion/frontend/frontend.py
|
||||||
|
chmod +x /docker/mount.sh /docker/run.sh
|
||||||
|
# streamlit
|
||||||
|
sed -i -- 's/8501/7860/g' /stable-diffusion/.streamlit/config.toml
|
||||||
|
EOF
|
||||||
|
|
||||||
|
WORKDIR /stable-diffusion
|
||||||
|
ENV PYTHONPATH="${PYTHONPATH}:${PWD}" STREAMLIT_SERVER_HEADLESS=true USE_STREAMLIT=0 CLI_ARGS=""
|
||||||
|
EXPOSE 7860
|
||||||
|
|
||||||
|
CMD /docker/mount.sh && /docker/run.sh
|
||||||
13
services/hlky/info.py
Normal file
13
services/hlky/info.py
Normal file
|
|
@ -0,0 +1,13 @@
|
||||||
|
import sys
|
||||||
|
from pathlib import Path
|
||||||
|
|
||||||
|
file = Path(sys.argv[1])
|
||||||
|
file.write_text(
|
||||||
|
file.read_text()\
|
||||||
|
.replace('<p>For help and advanced usage guides,', """
|
||||||
|
<p>
|
||||||
|
Created using <a href="https://github.com/AbdBarho/stable-diffusion-webui-docker">stable-diffusion-webui-docker</a>.
|
||||||
|
</p>
|
||||||
|
<p>For help and advanced usage guides,
|
||||||
|
""", 1)
|
||||||
|
)
|
||||||
32
services/hlky/mount.sh
Executable file
32
services/hlky/mount.sh
Executable file
|
|
@ -0,0 +1,32 @@
|
||||||
|
#!/bin/bash
|
||||||
|
|
||||||
|
set -Eeuo pipefail
|
||||||
|
|
||||||
|
declare -A MOUNTS
|
||||||
|
|
||||||
|
ROOT=/stable-diffusion/src
|
||||||
|
|
||||||
|
# cache
|
||||||
|
MOUNTS["/root/.cache"]=/data/.cache
|
||||||
|
# ui specific
|
||||||
|
MOUNTS["${PWD}/models/realesrgan"]=/data/RealESRGAN
|
||||||
|
MOUNTS["${PWD}/models/ldsr"]=/data/LDSR
|
||||||
|
MOUNTS["${PWD}/models/custom"]=/data/StableDiffusion
|
||||||
|
|
||||||
|
# hack
|
||||||
|
MOUNTS["${PWD}/models/gfpgan/GFPGANv1.3.pth"]=/data/GFPGAN/GFPGANv1.4.pth
|
||||||
|
MOUNTS["${PWD}/models/gfpgan/GFPGANv1.4.pth"]=/data/GFPGAN/GFPGANv1.4.pth
|
||||||
|
MOUNTS["${PWD}/gfpgan/weights"]=/data/.cache
|
||||||
|
|
||||||
|
|
||||||
|
for to_path in "${!MOUNTS[@]}"; do
|
||||||
|
set -Eeuo pipefail
|
||||||
|
from_path="${MOUNTS[${to_path}]}"
|
||||||
|
rm -rf "${to_path}"
|
||||||
|
mkdir -p "$(dirname "${to_path}")"
|
||||||
|
ln -sT "${from_path}" "${to_path}"
|
||||||
|
echo Mounted $(basename "${from_path}")
|
||||||
|
done
|
||||||
|
|
||||||
|
# streamlit config
|
||||||
|
ln -sf /docker/userconfig_streamlit.yaml /stable-diffusion/configs/webui/userconfig_streamlit.yaml
|
||||||
10
services/hlky/run.sh
Executable file
10
services/hlky/run.sh
Executable file
|
|
@ -0,0 +1,10 @@
|
||||||
|
#!/bin/bash
|
||||||
|
|
||||||
|
set -Eeuo pipefail
|
||||||
|
|
||||||
|
echo "USE_STREAMLIT = ${USE_STREAMLIT}"
|
||||||
|
if [ "${USE_STREAMLIT}" == "1" ]; then
|
||||||
|
python -u -m streamlit run scripts/webui_streamlit.py
|
||||||
|
else
|
||||||
|
python3 -u scripts/webui.py --outdir /output --ckpt /data/StableDiffusion/v1-5-pruned-emaonly.ckpt ${CLI_ARGS}
|
||||||
|
fi
|
||||||
9
services/hlky/userconfig_streamlit.yaml
Normal file
9
services/hlky/userconfig_streamlit.yaml
Normal file
|
|
@ -0,0 +1,9 @@
|
||||||
|
general:
|
||||||
|
version: 1.24.6
|
||||||
|
outdir: /output
|
||||||
|
default_model: "Stable Diffusion v1.5"
|
||||||
|
default_model_path: /data/StableDiffusion/v1-5-pruned-emaonly.ckpt
|
||||||
|
outdir_txt2img: /output/txt2img-samples
|
||||||
|
outdir_img2img: /output/img2img-samples
|
||||||
|
outdir_img2txt: /output/img2txt
|
||||||
|
optimized_turbo: True
|
||||||
50
services/lstein/Dockerfile
Normal file
50
services/lstein/Dockerfile
Normal file
|
|
@ -0,0 +1,50 @@
|
||||||
|
# syntax=docker/dockerfile:1
|
||||||
|
|
||||||
|
FROM continuumio/miniconda3:4.12.0
|
||||||
|
|
||||||
|
SHELL ["/bin/bash", "-ceuxo", "pipefail"]
|
||||||
|
|
||||||
|
ENV DEBIAN_FRONTEND=noninteractive PIP_EXISTS_ACTION=w PIP_PREFER_BINARY=1 PIP_NO_CACHE_DIR=1
|
||||||
|
|
||||||
|
# now it requires python3.9
|
||||||
|
RUN conda install python=3.9 && conda clean -a -y
|
||||||
|
RUN conda install pytorch==1.11.0 torchvision==0.12.0 cudatoolkit=11.3 -c pytorch && conda clean -a -y
|
||||||
|
|
||||||
|
RUN apt-get update && apt install fonts-dejavu-core -y && apt-get clean
|
||||||
|
|
||||||
|
RUN <<EOF
|
||||||
|
git clone https://github.com/invoke-ai/InvokeAI.git stable-diffusion
|
||||||
|
cd stable-diffusion
|
||||||
|
git reset --hard 986cc49d7a8441a70c709c04a55016a1c783f2ed
|
||||||
|
git config --global http.postBuffer 1048576000
|
||||||
|
# conda env update --file environment.yml -n base
|
||||||
|
conda clean -a -y
|
||||||
|
EOF
|
||||||
|
|
||||||
|
|
||||||
|
ARG BRANCH=development SHA=fdf9b1c40c243d6806e787877e42ae4c115790ee
|
||||||
|
RUN <<EOF
|
||||||
|
cd stable-diffusion
|
||||||
|
git fetch
|
||||||
|
git reset --hard
|
||||||
|
git checkout ${BRANCH}
|
||||||
|
git reset --hard ${SHA}
|
||||||
|
conda env update --file environment.yml -n base
|
||||||
|
conda clean -a -y
|
||||||
|
EOF
|
||||||
|
|
||||||
|
RUN pip uninstall opencv-python -y && pip install --prefer-binary --force-reinstall --no-cache-dir opencv-python-headless
|
||||||
|
|
||||||
|
COPY . /docker/
|
||||||
|
RUN <<EOF
|
||||||
|
python3 /docker/info.py /stable-diffusion/frontend/dist/index.html
|
||||||
|
chmod +x /docker/mount.sh
|
||||||
|
EOF
|
||||||
|
|
||||||
|
|
||||||
|
ENV PRELOAD=false CLI_ARGS=""
|
||||||
|
WORKDIR /stable-diffusion
|
||||||
|
EXPOSE 7860
|
||||||
|
|
||||||
|
CMD /docker/mount.sh && \
|
||||||
|
python3 -u scripts/invoke.py --outdir /output --web --host 0.0.0.0 --port 7860 ${CLI_ARGS}
|
||||||
13
services/lstein/info.py
Normal file
13
services/lstein/info.py
Normal file
|
|
@ -0,0 +1,13 @@
|
||||||
|
import sys
|
||||||
|
from pathlib import Path
|
||||||
|
|
||||||
|
file = Path(sys.argv[1])
|
||||||
|
file.write_text(
|
||||||
|
file.read_text()\
|
||||||
|
.replace(' <div id="root"></div>', """
|
||||||
|
<div id="root"></div>
|
||||||
|
<div>
|
||||||
|
Deployed with <a href="https://github.com/AbdBarho/stable-diffusion-webui-docker/">stable-diffusion-webui-docker</a>
|
||||||
|
</div>
|
||||||
|
""", 1)
|
||||||
|
)
|
||||||
29
services/lstein/mount.sh
Executable file
29
services/lstein/mount.sh
Executable file
|
|
@ -0,0 +1,29 @@
|
||||||
|
#!/bin/bash
|
||||||
|
|
||||||
|
set -Eeuo pipefail
|
||||||
|
|
||||||
|
declare -A MOUNTS
|
||||||
|
|
||||||
|
# cache
|
||||||
|
MOUNTS["/root/.cache"]=/data/.cache
|
||||||
|
# ui specific
|
||||||
|
MOUNTS["${PWD}/models/ldm/stable-diffusion-v1/model.ckpt"]=/data/StableDiffusion/model.ckpt
|
||||||
|
MOUNTS["${PWD}/src/gfpgan/experiments/pretrained_models/GFPGANv1.4.pth"]=/data/GFPGAN/GFPGANv1.4.pth
|
||||||
|
MOUNTS["${PWD}/ldm/invoke/restoration/codeformer/weights"]=/data/Codeformer
|
||||||
|
# hacks
|
||||||
|
MOUNTS["/opt/conda/lib/python3.9/site-packages/facexlib/weights"]=/data/.cache
|
||||||
|
MOUNTS["/opt/conda/lib/python3.9/site-packages/realesrgan/weights"]=/data/RealESRGAN
|
||||||
|
MOUNTS["${PWD}/gfpgan/weights"]=/data/.cache
|
||||||
|
|
||||||
|
for to_path in "${!MOUNTS[@]}"; do
|
||||||
|
set -Eeuo pipefail
|
||||||
|
from_path="${MOUNTS[${to_path}]}"
|
||||||
|
rm -rf "${to_path}"
|
||||||
|
mkdir -p "$(dirname "${to_path}")"
|
||||||
|
ln -sT "${from_path}" "${to_path}"
|
||||||
|
echo Mounted $(basename "${from_path}")
|
||||||
|
done
|
||||||
|
|
||||||
|
if "${PRELOAD}" == "true"; then
|
||||||
|
python3 -u scripts/preload_models.py
|
||||||
|
fi
|
||||||
Loading…
Reference in a new issue