mirror of
https://github.com/AbdBarho/stable-diffusion-webui-docker.git
synced 2025-10-27 16:24:26 -04:00
Compare commits
62 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
14c4b36aff | ||
|
|
28f171e64d | ||
|
|
9af4a23ec4 | ||
|
|
24ecd676ab | ||
|
|
ef36c50cf9 | ||
|
|
43a5e5e85f | ||
|
|
5bbc21ea3d | ||
|
|
09366ed955 | ||
|
|
d4874e7c3a | ||
|
|
7638fb4e5e | ||
|
|
15a61a99d6 | ||
|
|
556a50f49b | ||
|
|
b899f4e516 | ||
|
|
a8c85b4699 | ||
|
|
a96285d10b | ||
|
|
83b78fe504 | ||
|
|
84f9cb84e7 | ||
|
|
6a66ff6abb | ||
|
|
59892da866 | ||
|
|
fceb83c2b0 | ||
|
|
17b01a7627 | ||
|
|
b96d7c30d0 | ||
|
|
aae83bb8f2 | ||
|
|
10763a8f61 | ||
|
|
64e8f093d2 | ||
|
|
3e0a137c23 | ||
|
|
a1c16942ff | ||
|
|
6ae3473214 | ||
|
|
5d731cb43c | ||
|
|
c1fa2f1457 | ||
|
|
d8cfdd3af5 | ||
|
|
03d12cbcd9 | ||
|
|
2e76b6c4e7 | ||
|
|
5eae2076ce | ||
|
|
725e1f39ba | ||
|
|
ab651fe0d7 | ||
|
|
f76f8d4671 | ||
|
|
e32a48f42a | ||
|
|
76989b39a6 | ||
|
|
4d9fc381bb | ||
|
|
bcee253fe0 | ||
|
|
499143009a | ||
|
|
c614625f04 | ||
|
|
ccd6e238b2 | ||
|
|
829864af9b | ||
|
|
ccc7306f48 | ||
|
|
082876aab3 | ||
|
|
ae834cb764 | ||
|
|
5f6d9fbb03 | ||
|
|
d4da252343 | ||
|
|
5af482ed8c | ||
|
|
ce4e190f8f | ||
|
|
bae3590980 | ||
|
|
1588d1eecf | ||
|
|
9cbd58b3f4 | ||
|
|
089fc524d8 | ||
|
|
0d8b7d4ac8 | ||
|
|
561664ea6e | ||
|
|
77c2b2d217 | ||
|
|
6c0c610f27 | ||
|
|
dc730b7f6b | ||
|
|
15952906a1 |
5
.devscripts/chmod.sh
Executable file
5
.devscripts/chmod.sh
Executable file
@@ -0,0 +1,5 @@
|
||||
#!/bin/bash
|
||||
|
||||
set -Eeuo pipefail
|
||||
|
||||
find . -name "*.sh" -exec git update-index --chmod=+x {} \;
|
||||
9
.editorconfig
Normal file
9
.editorconfig
Normal file
@@ -0,0 +1,9 @@
|
||||
root = true
|
||||
|
||||
[*]
|
||||
end_of_line = lf
|
||||
indent_style = space
|
||||
indent_size = 2
|
||||
charset = utf-8
|
||||
insert_final_newline = true
|
||||
trim_trailing_whitespace = true
|
||||
1
.gitattributes
vendored
Normal file
1
.gitattributes
vendored
Normal file
@@ -0,0 +1 @@
|
||||
* text=auto eol=lf
|
||||
36
.github/ISSUE_TEMPLATE/bug.md
vendored
Normal file
36
.github/ISSUE_TEMPLATE/bug.md
vendored
Normal file
@@ -0,0 +1,36 @@
|
||||
---
|
||||
name: Bug
|
||||
about: Report a bug
|
||||
title: ''
|
||||
labels: bug
|
||||
assignees: ''
|
||||
|
||||
---
|
||||
|
||||
**Has this issue been opened before? Check the [FAQ](https://github.com/AbdBarho/stable-diffusion-webui-docker/wiki/Main), the [issues](https://github.com/AbdBarho/stable-diffusion-webui-docker/issues?q=is%3Aissue)**
|
||||
|
||||
|
||||
|
||||
**Describe the bug**
|
||||
|
||||
|
||||
**Which UI**
|
||||
|
||||
hlky or auto or auto-cpu or lstein?
|
||||
|
||||
**Steps to Reproduce**
|
||||
1. Go to '...'
|
||||
2. Click on '....'
|
||||
3. Scroll down to '....'
|
||||
4. See error
|
||||
|
||||
**Hardware / Software:**
|
||||
- OS: [e.g. Windows / Ubuntu and version]
|
||||
- RAM:
|
||||
- GPU: [Nvidia 1660 / No GPU]
|
||||
- VRAM:
|
||||
- Docker Version, Docker compose version
|
||||
- Release version [e.g. 1.0.1]
|
||||
|
||||
**Additional context**
|
||||
Any other context about the problem here. If applicable, add screenshots to help explain your problem.
|
||||
5
.github/pull_request_template.md
vendored
Normal file
5
.github/pull_request_template.md
vendored
Normal file
@@ -0,0 +1,5 @@
|
||||
### Update versions
|
||||
|
||||
- auto: https://github.com/AUTOMATIC1111/stable-diffusion-webui/commit/
|
||||
- hlky: https://github.com/sd-webui/stable-diffusion-webui/commit/
|
||||
- lstein: https://github.com/lstein/stable-diffusion/commit/
|
||||
19
.github/workflows/docker.yml
vendored
Normal file
19
.github/workflows/docker.yml
vendored
Normal file
@@ -0,0 +1,19 @@
|
||||
name: Build Images
|
||||
|
||||
on: [push]
|
||||
|
||||
jobs:
|
||||
build:
|
||||
strategy:
|
||||
matrix:
|
||||
profile:
|
||||
- auto
|
||||
- hlky
|
||||
- lstein
|
||||
- download
|
||||
runs-on: ubuntu-latest
|
||||
name: ${{ matrix.profile }}
|
||||
steps:
|
||||
- uses: actions/checkout@v3
|
||||
# better caching?
|
||||
- run: docker compose --profile ${{ matrix.profile }} build --progress plain
|
||||
22
.github/workflows/executable.yml1
vendored
Normal file
22
.github/workflows/executable.yml1
vendored
Normal file
@@ -0,0 +1,22 @@
|
||||
name: Check executable
|
||||
|
||||
on: [push]
|
||||
|
||||
jobs:
|
||||
check:
|
||||
runs-on: ubuntu-latest
|
||||
name: Check all sh
|
||||
steps:
|
||||
- run: git config --global core.fileMode true
|
||||
- uses: actions/checkout@v3
|
||||
- shell: bash
|
||||
run: |
|
||||
shopt -s globstar;
|
||||
FAIL=0
|
||||
for file in **/*.sh; do
|
||||
if [ -f "${file}" ] && [ -r "${file}" ] && [ ! -x "${file}" ]; then
|
||||
echo "$file" is not executable;
|
||||
FAIL=1
|
||||
fi
|
||||
done
|
||||
exit ${FAIL}
|
||||
20
.github/workflows/stale.yml
vendored
Normal file
20
.github/workflows/stale.yml
vendored
Normal file
@@ -0,0 +1,20 @@
|
||||
name: 'Close stale issues and PRs'
|
||||
on:
|
||||
schedule:
|
||||
- cron: '30 1 * * *'
|
||||
|
||||
jobs:
|
||||
stale:
|
||||
runs-on: ubuntu-latest
|
||||
steps:
|
||||
- uses: actions/stale@v5
|
||||
with:
|
||||
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-pr-message: This PR 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.
|
||||
close-issue-message: This issue was closed because it has been stalled for 7 days with no activity.
|
||||
close-pr-message: This PR was closed because it has been stalled for 7 days with no activity.
|
||||
days-before-issue-stale: 14
|
||||
days-before-pr-stale: 14
|
||||
days-before-issue-close: 7
|
||||
days-before-pr-close: 7
|
||||
2
.gitignore
vendored
2
.gitignore
vendored
@@ -1,2 +1,4 @@
|
||||
/dev
|
||||
/.devcontainer
|
||||
embeddings/*
|
||||
/docker-compose.override.yml
|
||||
|
||||
9
LICENSE
9
LICENSE
@@ -86,4 +86,11 @@ administration of justice, law enforcement, immigration or asylum
|
||||
processes, such as predicting an individual will commit fraud/crime
|
||||
commitment (e.g. by text profiling, drawing causal relationships between
|
||||
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/TencentARC/GFPGAN/blob/master/LICENSE
|
||||
https://github.com/xinntao/Real-ESRGAN/blob/master/LICENSE
|
||||
|
||||
138
README.md
138
README.md
@@ -1,61 +1,77 @@
|
||||
# Stable Diffusion WebUI Docker
|
||||
|
||||
Run Stable Diffusion on your machine with a nice UI without any hassle!
|
||||
|
||||
This repository provides the [WebUI](https://github.com/hlky/stable-diffusion-webui) as docker for easy setup and deployment. Please note that this repo delivers all cutting-edge unstable changes from the WebUI, so expect some bugs.
|
||||
|
||||
## Setup
|
||||
|
||||
make sure you have docker installed and up to date. Download this repo and run:
|
||||
|
||||
```
|
||||
docker compose build
|
||||
```
|
||||
|
||||
you can let it build in the background while you download the different models
|
||||
|
||||
- [Stable Diffusion v1.4 (4GB)](https://www.googleapis.com/storage/v1/b/aai-blog-files/o/sd-v1-4.ckpt?alt=media), rename to `model.ckpt`
|
||||
- (Optional) [GFPGANv1.3.pth (333MB)](https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.3.pth) to improve generated faces.
|
||||
- (Optional) [RealESRGAN_x4plus.pth (64MB)](https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.0/RealESRGAN_x4plus.pth) and [RealESRGAN_x4plus_anime_6B.pth (18MB)](https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.2.4/RealESRGAN_x4plus_anime_6B.pth) for super-sampling.
|
||||
|
||||
Put all of the downloaded files in the `models` folder, it should look something like this:
|
||||
|
||||
```
|
||||
models/
|
||||
├── GFPGANv1.3.pth
|
||||
├── RealESRGAN_x4plus.pth
|
||||
├── RealESRGAN_x4plus_anime_6B.pth
|
||||
└── model.ckpt
|
||||
```
|
||||
|
||||
## Run
|
||||
|
||||
After the build is done, you can run the app with:
|
||||
|
||||
```
|
||||
docker compose up --build
|
||||
```
|
||||
|
||||
Will start the app on http://localhost:7860/
|
||||
|
||||
Note: the first start will take sometime as some other models will be downloaded, these will be cached in the `cache` folder, so next runs are faster.
|
||||
|
||||
## Config
|
||||
|
||||
in the `docker-compose.yml` you can change the `CLI_ARGS` variable contains all of the variables that will be passed to [the web ui](https://github.com/hlky/stable-diffusion/blob/fa977b3d6f9d0b264035c949fd70415476f00036/scripts/webui.py).
|
||||
|
||||
By default: `--extra-models-cpu --optimized-turbo` are given, which allow you to use this model on a 6GB GPU.
|
||||
|
||||
# Disclaimer
|
||||
|
||||
The authors of this project are not responsible for any content generated using this interface.
|
||||
|
||||
This license of this software forbids you from sharing any content that violates any laws, produce any harm to a person, disseminate any personal information that would be meant for harm, spread misinformation and target vulnerable groups. For the full list of restrictions please read [the license](./LICENSE).
|
||||
|
||||
# Thanks
|
||||
|
||||
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)
|
||||
- [CompVis/stable-diffusion](https://github.com/CompVis/stable-diffusion)
|
||||
# Stable Diffusion WebUI Docker
|
||||
|
||||
Run Stable Diffusion on your machine with a nice UI without any hassle!
|
||||
|
||||
This repository provides multiple UIs for you to play around with stable diffusion:
|
||||
|
||||
## Features
|
||||
|
||||
### AUTOMATIC1111
|
||||
|
||||
[AUTOMATIC1111's fork](https://github.com/AUTOMATIC1111/stable-diffusion-webui) is imho the most feature rich yet elegant UI:
|
||||
|
||||
- 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 |
|
||||
| ---------------------------------------------------------------------------------------------------------- | ---------------------------------------------------------------------------------------------------------- | ---------------------------------------------------------------------------------------------------------- |
|
||||
|  |  |  |
|
||||
|
||||
### hlky
|
||||
|
||||
[hlky's fork](https://github.com/hlky/stable-diffusion-webui) is one of the most popular UIs, with many features:
|
||||
|
||||
- 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/sd-webui/stable-diffusion-webui/blob/master/README.md)
|
||||
|
||||
Screenshots:
|
||||
|
||||
| Text to image | Image to image | Image Lab |
|
||||
| ---------------------------------------------------------------------------------------------------------- | ---------------------------------------------------------------------------------------------------------- | ---------------------------------------------------------------------------------------------------------- |
|
||||
|  |  |  |
|
||||
|
||||
<!--
|
||||
### lstein
|
||||
|
||||
[lstein's fork](https://github.com/lstein/stable-diffusion) is very mature when it comes to the cli, and the WebUI has potential.
|
||||
|
||||
| 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
|
||||
|
||||
Contributions are welcome! create an issue first of what you want to contribute (before you implement anything) so we can talk about it.
|
||||
|
||||
## Disclaimer
|
||||
|
||||
The authors of this project are not responsible for any content generated using this interface.
|
||||
|
||||
This license of this software forbids you from sharing any content that violates any laws, produce any harm to a person, disseminate any personal information that would be meant for harm, spread misinformation and target vulnerable groups. For the full list of restrictions please read [the license](./LICENSE).
|
||||
|
||||
## Thanks
|
||||
|
||||
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)
|
||||
- [lstein/stable-diffusion](https://github.com/lstein/stable-diffusion)
|
||||
- [CompVis/stable-diffusion](https://github.com/CompVis/stable-diffusion)
|
||||
- [hlky/sd-enable-textual-inversion](https://github.com/hlky/sd-enable-textual-inversion)
|
||||
- [devilismyfriend/latent-diffusion](https://github.com/devilismyfriend/latent-diffusion)
|
||||
- [Hafiidz/latent-diffusion](https://github.com/Hafiidz/latent-diffusion)
|
||||
|
||||
@@ -1,57 +0,0 @@
|
||||
# syntax=docker/dockerfile:1
|
||||
|
||||
FROM continuumio/miniconda3:4.12.0
|
||||
|
||||
|
||||
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 git clone https://github.com/hlky/stable-diffusion.git && cd stable-diffusion && git reset --hard ff8c2d0b709f1e4180fb19fa5c27ec28c414cedd
|
||||
RUN conda env update --file stable-diffusion/environment.yaml --name base && conda clean -a -y
|
||||
|
||||
# fonts for generating the grid
|
||||
RUN apt-get update && apt install fonts-dejavu-core && apt-get clean
|
||||
|
||||
# Note: don't update the sha of previous versions because the install will take forever
|
||||
# instead, update the repo state in a later step
|
||||
# RUN cd stable-diffusion && git pull && git reset --hard ff8c2d0b709f1e4180fb19fa5c27ec28c414cedd && \
|
||||
# conda env update --file environment.yaml --name base && conda clean -a -y
|
||||
|
||||
# download dev UI version, update the sha below in case you want some other version
|
||||
# RUN <<EOF
|
||||
# git clone https://github.com/hlky/stable-diffusion-webui.git
|
||||
# cd stable-diffusion-webui
|
||||
# # map to this file: https://github.com/hlky/stable-diffusion-webui/blob/master/.github/sync.yml
|
||||
# git reset --hard 49e6178fd82ca736f9bbc621c6b12487c300e493
|
||||
# cp -t /stable-diffusion/scripts/ webui.py relauncher.py txt2img.yaml
|
||||
# cp -t /stable-diffusion/configs/webui webui.yaml
|
||||
# cp -t /stable-diffusion/frontend/ frontend/*
|
||||
# cd / && rm -rf stable-diffusion-webui
|
||||
# EOF
|
||||
|
||||
# Textual-inversion:
|
||||
# RUN <<EOF
|
||||
# git clone https://github.com/hlky/sd-enable-textual-inversion.git
|
||||
# cp -rf sd-enable-textual-inversion /stable-diffusion
|
||||
# EOF
|
||||
|
||||
# add info
|
||||
COPY info.py /info.py
|
||||
RUN python /info.py /stable-diffusion/frontend/frontend.py
|
||||
|
||||
WORKDIR /stable-diffusion
|
||||
ENV TRANSFORMERS_CACHE=/cache/transformers TORCH_HOME=/cache/torch CLI_ARGS="" \
|
||||
GFPGAN_PATH=/stable-diffusion/src/gfpgan/experiments/pretrained_models/GFPGANv1.3.pth \
|
||||
RealESRGAN_PATH=/stable-diffusion/src/realesrgan/experiments/pretrained_models/RealESRGAN_x4plus.pth \
|
||||
RealESRGAN_ANIME_PATH=/stable-diffusion/src/realesrgan/experiments/pretrained_models/RealESRGAN_x4plus_anime_6B.pth
|
||||
EXPOSE 7860
|
||||
CMD \
|
||||
for path in "${GFPGAN_PATH}" "${RealESRGAN_PATH}" "${RealESRGAN_ANIME_PATH}"; do \
|
||||
name=$(basename "${path}"); \
|
||||
base=$(dirname "${path}"); \
|
||||
test -f "/models/${name}" && mkdir -p "${base}" && ln -sf "/models/${name}" "${path}" && echo "Mounted ${name}";\
|
||||
done;\
|
||||
# force facexlib cache
|
||||
mkdir -p /cache/weights/ && rm -rf /stable-diffusion/src/facexlib/facexlib/weights && \
|
||||
ln -sf /cache/weights/ /stable-diffusion/src/facexlib/facexlib/ && \
|
||||
# run, -u to not buffer stdout / stderr
|
||||
python3 -u scripts/webui.py --outdir /output --ckpt /models/model.ckpt --save-metadata ${CLI_ARGS}
|
||||
3
cache/.gitignore
vendored
3
cache/.gitignore
vendored
@@ -1,3 +0,0 @@
|
||||
/torch
|
||||
/transformers
|
||||
/weights
|
||||
13
data/.gitignore
vendored
Normal file
13
data/.gitignore
vendored
Normal file
@@ -0,0 +1,13 @@
|
||||
# for all of the stuff downloaded by transformers, pytorch, and others
|
||||
/.cache
|
||||
# for all stable diffusion models (main, waifu diffusion, etc..)
|
||||
/StableDiffusion
|
||||
# others
|
||||
/Codeformer
|
||||
/GFPGAN
|
||||
/ESRGAN
|
||||
/BSRGAN
|
||||
/RealESRGAN
|
||||
/SwinIR
|
||||
/LDSR
|
||||
# TODO: add the embeddings folder here
|
||||
@@ -1,17 +1,11 @@
|
||||
version: '3.9'
|
||||
|
||||
services:
|
||||
model:
|
||||
build: ./build/
|
||||
restart: on-failure
|
||||
x-base_service: &base_service
|
||||
ports:
|
||||
- "7860:7860"
|
||||
volumes:
|
||||
- ./cache:/cache
|
||||
- ./output:/output
|
||||
- ./models:/models
|
||||
environment:
|
||||
- CLI_ARGS=--extra-models-cpu --optimized-turbo
|
||||
- &v1 ./data:/data
|
||||
- &v2 ./output:/output
|
||||
deploy:
|
||||
resources:
|
||||
reservations:
|
||||
@@ -19,3 +13,45 @@ services:
|
||||
- driver: nvidia
|
||||
device_ids: ['0']
|
||||
capabilities: [gpu]
|
||||
|
||||
name: webui-docker
|
||||
|
||||
services:
|
||||
download:
|
||||
build: ./services/download/
|
||||
profiles: ["download"]
|
||||
volumes:
|
||||
- *v1
|
||||
|
||||
hlky:
|
||||
<<: *base_service
|
||||
profiles: ["hlky"]
|
||||
build: ./services/hlky/
|
||||
environment:
|
||||
- CLI_ARGS=--optimized-turbo
|
||||
|
||||
auto: &automatic
|
||||
<<: *base_service
|
||||
profiles: ["auto"]
|
||||
build: ./services/AUTOMATIC1111
|
||||
volumes:
|
||||
- *v1
|
||||
- *v2
|
||||
- ./services/AUTOMATIC1111/config.json:/stable-diffusion-webui/config.json
|
||||
environment:
|
||||
- CLI_ARGS=--allow-code --medvram
|
||||
|
||||
auto-cpu:
|
||||
<<: *automatic
|
||||
profiles: ["auto-cpu"]
|
||||
deploy: {}
|
||||
environment:
|
||||
- CLI_ARGS=--no-half --precision full
|
||||
|
||||
lstein:
|
||||
<<: *base_service
|
||||
profiles: ["lstein"]
|
||||
build: ./services/lstein/
|
||||
environment:
|
||||
- PRELOAD=false
|
||||
- CLI_ARGS=
|
||||
|
||||
4
models/.gitignore
vendored
4
models/.gitignore
vendored
@@ -1,4 +0,0 @@
|
||||
/model.ckpt
|
||||
/GFPGANv1.3.pth
|
||||
/RealESRGAN_x4plus.pth
|
||||
/RealESRGAN_x4plus_anime_6B.pth
|
||||
30
scripts/migratev1tov2.sh
Executable file
30
scripts/migratev1tov2.sh
Executable file
@@ -0,0 +1,30 @@
|
||||
mkdir -p data/.cache data/StableDiffusion data/Codeformer data/GFPGAN data/ESRGAN data/BSRGAN data/RealESRGAN data/SwinIR data/LDSR /data/embeddings
|
||||
|
||||
cp -vf cache/models/model.ckpt data/StableDiffusion/model.ckpt
|
||||
|
||||
cp -vf cache/models/LDSR.ckpt data/LDSR/model.ckpt
|
||||
cp -vf cache/models/LDSR.yaml data/LDSR/project.yaml
|
||||
|
||||
cp -vf cache/models/RealESRGAN_x4plus.pth data/RealESRGAN/
|
||||
cp -vf cache/models/RealESRGAN_x4plus_anime_6B.pth data/RealESRGAN/
|
||||
|
||||
cp -vrf cache/torch data/.cache/
|
||||
|
||||
mkdir -p data/.cache/huggingface/transformers/
|
||||
cp -vrf cache/transformers/* data/.cache/huggingface/transformers/
|
||||
|
||||
cp -v cache/custom-models/* data/StableDiffusion/
|
||||
|
||||
mkdir -p data/.cache/clip/
|
||||
cp -vf cache/weights/ViT-L-14.pt data/.cache/clip/
|
||||
|
||||
cp -vf cache/weights/codeformer.pth data/Codeformer/codeformer-v0.1.0.pth
|
||||
|
||||
cp -vf cache/weights/detection_Resnet50_Final.pth data/.cache/
|
||||
cp -vf cache/weights/parsing_parsenet.pth data/.cache/
|
||||
|
||||
cp -v embeddings/* data/embeddings/
|
||||
|
||||
echo this script was created 10/2022
|
||||
echo Dont forget to run: docker compose --profile download up --build
|
||||
echo the cache and embeddings folders can be deleted, but its not necessary.
|
||||
75
services/AUTOMATIC1111/Dockerfile
Normal file
75
services/AUTOMATIC1111/Dockerfile
Normal file
@@ -0,0 +1,75 @@
|
||||
# syntax=docker/dockerfile:1
|
||||
|
||||
FROM alpine/git:2.36.2 as download
|
||||
|
||||
|
||||
RUN git clone https://github.com/CompVis/stable-diffusion.git repositories/stable-diffusion && cd repositories/stable-diffusion && git reset --hard 69ae4b35e0a0f6ee1af8bb9a5d0016ccb27e36dc
|
||||
|
||||
RUN git clone https://github.com/sczhou/CodeFormer.git repositories/CodeFormer && cd repositories/CodeFormer && git reset --hard c5b4593074ba6214284d6acd5f1719b6c5d739af
|
||||
RUN git clone https://github.com/salesforce/BLIP.git repositories/BLIP && cd repositories/BLIP && git reset --hard 48211a1594f1321b00f14c9f7a5b4813144b2fb9
|
||||
RUN git clone https://github.com/Hafiidz/latent-diffusion.git repositories/latent-diffusion && cd repositories/latent-diffusion && git reset --hard abf33e7002d59d9085081bce93ec798dcabd49af
|
||||
|
||||
RUN <<EOF
|
||||
# because taming-transformers is huge
|
||||
git config --global http.postBuffer 1048576000
|
||||
git clone https://github.com/CompVis/taming-transformers.git repositories/taming-transformers
|
||||
git reset --hard 24268930bf1dce879235a7fddd0b2355b84d7ea6
|
||||
rm -rf repositories/taming-transformers/data repositories/taming-transformers/assets
|
||||
EOF
|
||||
|
||||
RUN git clone https://github.com/crowsonkb/k-diffusion.git repositories/k-diffusion && cd repositories/k-diffusion && git reset --hard 9e3002b7cd64df7870e08527b7664eb2f2f5f3f5
|
||||
|
||||
|
||||
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 -y && apt-get clean
|
||||
|
||||
|
||||
RUN <<EOF
|
||||
git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
|
||||
cd stable-diffusion-webui
|
||||
git reset --hard 15f333a266c20319e2b95a47a8834adf7b914aec
|
||||
pip install --prefer-binary --no-cache-dir -r requirements_versions.txt
|
||||
EOF
|
||||
|
||||
ENV ROOT=/stable-diffusion-webui \
|
||||
WORKDIR=/stable-diffusion-webui/repositories/stable-diffusion
|
||||
|
||||
|
||||
COPY --from=download /git/ ${ROOT}
|
||||
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=3f417566b0bda8eab05d247567aebf001c1d1725
|
||||
RUN <<EOF
|
||||
cd stable-diffusion-webui
|
||||
git pull --rebase
|
||||
git reset --hard ${SHA}
|
||||
pip install --prefer-binary --no-cache-dir -r requirements_versions.txt
|
||||
pip install --prefer-binary --no-cache-dir -r requirements.txt
|
||||
EOF
|
||||
|
||||
RUN pip install --prefer-binary -U --no-cache-dir opencv-python-headless
|
||||
|
||||
|
||||
COPY . /docker
|
||||
RUN <<EOF
|
||||
chmod +x /docker/mount.sh && python3 /docker/info.py ${ROOT}/modules/ui.py
|
||||
EOF
|
||||
|
||||
ENV CLI_ARGS=""
|
||||
WORKDIR ${WORKDIR}
|
||||
EXPOSE 7860
|
||||
# run, -u to not buffer stdout / stderr
|
||||
CMD /docker/mount.sh && \
|
||||
python3 -u ../../webui.py --listen --port 7860 --hide-ui-dir-config --ckpt-dir ${ROOT}/models/Stable-diffusion ${CLI_ARGS}
|
||||
# --gfpgan-models-path /data/GFPGAN --codeformer-models-path /data/Codeformer --esrgan-models-path /data/ESRGAN --bsrgan-models-path /data/BSRGAN --realesrgan-models-path /data/RealESRGAN --swinir-models-path /data/SwinIR --ldsr-models-path /data/LDSR
|
||||
79
services/AUTOMATIC1111/config.json
Normal file
79
services/AUTOMATIC1111/config.json
Normal file
@@ -0,0 +1,79 @@
|
||||
{
|
||||
"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",
|
||||
"__WARNING__": "DON'T CHANGE ANYTHING BEFORE THIS",
|
||||
|
||||
"add_model_hash_to_info": false,
|
||||
"code_former_weight": 0.5,
|
||||
"directories_filename_pattern": "",
|
||||
"directories_max_prompt_words": 8,
|
||||
"enable_batch_seeds": true,
|
||||
"enable_emphasis": true,
|
||||
"enable_pnginfo": true,
|
||||
"enable_quantization": false,
|
||||
"ESRGAN_tile": 192,
|
||||
"ESRGAN_tile_overlap": 8,
|
||||
"export_for_4chan": true,
|
||||
"face_restoration_model": null,
|
||||
"face_restoration_unload": false,
|
||||
"filter_nsfw": false,
|
||||
"grid_extended_filename": false,
|
||||
"grid_format": "png",
|
||||
"grid_only_if_multiple": true,
|
||||
"grid_save": true,
|
||||
"grid_save_to_dirs": false,
|
||||
"img2img_color_correction": false,
|
||||
"img2img_fix_steps": false,
|
||||
"interrogate_clip_dict_limit": 1500,
|
||||
"interrogate_clip_max_length": 48,
|
||||
"interrogate_clip_min_length": 24,
|
||||
"interrogate_clip_num_beams": 1,
|
||||
"interrogate_keep_models_in_memory": false,
|
||||
"interrogate_use_builtin_artists": true,
|
||||
"jpeg_quality": 80,
|
||||
"js_modal_lightbox": true,
|
||||
"js_modal_lightbox_initialy_zoomed": true,
|
||||
"ldsr_post_down": 1,
|
||||
"ldsr_pre_down": 1,
|
||||
"ldsr_steps": 30,
|
||||
"memmon_poll_rate": 8,
|
||||
"multiple_tqdm": true,
|
||||
"n_rows": -1,
|
||||
"outdir_grids": "",
|
||||
"random_artist_categories": [],
|
||||
"realesrgan_enabled_models": [
|
||||
"Real-ESRGAN 4x plus",
|
||||
"Real-ESRGAN 4x plus anime 6B",
|
||||
"Real-ESRGAN 2x plus",
|
||||
"Real-ESRGAN AnimeVideo",
|
||||
"Real-ESRGAN General WDN x4x3",
|
||||
"Real-ESRGAN General x4x3"
|
||||
],
|
||||
"return_grid": true,
|
||||
"samples_filename_format": "",
|
||||
"samples_filename_pattern": "",
|
||||
"samples_format": "png",
|
||||
"samples_log_stdout": false,
|
||||
"samples_save": true,
|
||||
"save_images_before_color_correction": false,
|
||||
"save_images_before_face_restoration": false,
|
||||
"save_selected_only": false,
|
||||
"save_to_dirs": false,
|
||||
"save_to_dirs_prompt_len": 10,
|
||||
"save_txt": false,
|
||||
"sd_model_checkpoint": null,
|
||||
"show_progress_every_n_steps": 7,
|
||||
"show_progressbar": true,
|
||||
"SWIN_tile": 192,
|
||||
"SWIN_tile_overlap": 8,
|
||||
"upscale_at_full_resolution_padding": 16,
|
||||
"upscaler_for_hires_fix": null,
|
||||
"upscaler_for_img2img": null,
|
||||
"use_original_name_batch": false
|
||||
}
|
||||
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)
|
||||
)
|
||||
31
services/AUTOMATIC1111/mount.sh
Executable file
31
services/AUTOMATIC1111/mount.sh
Executable file
@@ -0,0 +1,31 @@
|
||||
#!/bin/bash
|
||||
|
||||
set -Eeuo pipefail
|
||||
|
||||
declare -A MOUNTS
|
||||
|
||||
MOUNTS["/root/.cache"]="/data/.cache"
|
||||
|
||||
# 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/LDSR"]="/data/LDSR"
|
||||
|
||||
MOUNTS["${ROOT}/embeddings"]="/data/embeddings"
|
||||
|
||||
# extra hacks
|
||||
MOUNTS["${ROOT}/repositories/CodeFormer/weights/facelib"]="/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
|
||||
6
services/download/Dockerfile
Normal file
6
services/download/Dockerfile
Normal file
@@ -0,0 +1,6 @@
|
||||
FROM bash:alpine3.15
|
||||
|
||||
RUN apk add parallel aria2
|
||||
COPY . /docker
|
||||
RUN chmod +x /docker/download.sh
|
||||
ENTRYPOINT ["/docker/download.sh"]
|
||||
6
services/download/checksums.sha256
Normal file
6
services/download/checksums.sha256
Normal file
@@ -0,0 +1,6 @@
|
||||
fe4efff1e174c627256e44ec2991ba279b3816e364b49f9be2abc0b3ff3f8556 /data/StableDiffusion/model.ckpt
|
||||
e2cd4703ab14f4d01fd1383a8a8b266f9a5833dacee8e6a79d3bf21a1b6be5ad /data/GFPGAN/GFPGANv1.4.pth
|
||||
4fa0d38905f75ac06eb49a7951b426670021be3018265fd191d2125df9d682f1 /data/RealESRGAN/RealESRGAN_x4plus.pth
|
||||
f872d837d3c90ed2e05227bed711af5671a6fd1c9f7d7e91c911a61f155e99da /data/RealESRGAN/RealESRGAN_x4plus_anime_6B.pth
|
||||
c209caecac2f97b4bb8f4d726b70ac2ac9b35904b7fc99801e1f5e61f9210c13 /data/LDSR/model.ckpt
|
||||
9d6ad53c5dafeb07200fb712db14b813b527edd262bc80ea136777bdb41be2ba /data/LDSR/project.yaml
|
||||
23
services/download/download.sh
Executable file
23
services/download/download.sh
Executable file
@@ -0,0 +1,23 @@
|
||||
#!/usr/bin/env bash
|
||||
|
||||
set -Eeuo pipefail
|
||||
|
||||
mkdir -p /data/.cache /data/StableDiffusion /data/Codeformer /data/GFPGAN /data/ESRGAN /data/BSRGAN /data/RealESRGAN /data/SwinIR /data/LDSR
|
||||
|
||||
cat <<EOF
|
||||
By using this software, you agree to the following licenses:
|
||||
https://github.com/CompVis/stable-diffusion/blob/main/LICENSE
|
||||
https://github.com/TencentARC/GFPGAN/blob/master/LICENSE
|
||||
https://github.com/xinntao/Real-ESRGAN/blob/master/LICENSE
|
||||
EOF
|
||||
|
||||
echo "Downloading, this might take a while..."
|
||||
|
||||
aria2c --input-file /docker/links.txt --dir /data --continue
|
||||
|
||||
echo "Checking SHAs..."
|
||||
|
||||
parallel --will-cite -a /docker/checksums.sha256 "echo -n {} | sha256sum -c"
|
||||
|
||||
# fix potential permissions
|
||||
chmod -R 777 /data /output
|
||||
12
services/download/links.txt
Normal file
12
services/download/links.txt
Normal file
@@ -0,0 +1,12 @@
|
||||
https://www.googleapis.com/storage/v1/b/aai-blog-files/o/sd-v1-4.ckpt?alt=media
|
||||
out=StableDiffusion/model.ckpt
|
||||
https://github.com/TencentARC/GFPGAN/releases/download/v1.3.4/GFPGANv1.4.pth
|
||||
out=GFPGAN/GFPGANv1.4.pth
|
||||
https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.0/RealESRGAN_x4plus.pth
|
||||
out=RealESRGAN/RealESRGAN_x4plus.pth
|
||||
https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.2.4/RealESRGAN_x4plus_anime_6B.pth
|
||||
out=RealESRGAN/RealESRGAN_x4plus_anime_6B.pth
|
||||
https://heibox.uni-heidelberg.de/f/31a76b13ea27482981b4/?dl=1
|
||||
out=LDSR/project.yaml
|
||||
https://heibox.uni-heidelberg.de/f/578df07c8fc04ffbadf3/?dl=1
|
||||
out=LDSR/model.ckpt
|
||||
50
services/hlky/Dockerfile
Normal file
50
services/hlky/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
|
||||
|
||||
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
|
||||
|
||||
|
||||
RUN <<EOF
|
||||
git config --global http.postBuffer 1048576000
|
||||
git clone https://github.com/sd-webui/stable-diffusion-webui.git stable-diffusion
|
||||
cd stable-diffusion
|
||||
git reset --hard 7623a5734740025d79b710f3744bff9276e1467b
|
||||
conda env update --file environment.yaml -n base
|
||||
conda clean -a -y
|
||||
EOF
|
||||
|
||||
|
||||
# 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 BRANCH=master SHA=d0bb60a139d60e6c2b9be4e18e0e29a86aa5af59
|
||||
ARG BRANCH=dev SHA=aaa3be16e0e66f07c63dbdb49b9b7cac8fef5cfb
|
||||
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
|
||||
|
||||
RUN pip install -U --no-cache-dir pyperclip
|
||||
|
||||
# add info
|
||||
COPY . /docker/
|
||||
RUN python /docker/info.py /stable-diffusion/frontend/frontend.py && chmod +x /docker/mount.sh
|
||||
|
||||
WORKDIR /stable-diffusion
|
||||
ENV PYTHONPATH="${PYTHONPATH}:${PWD}" CLI_ARGS=""
|
||||
EXPOSE 7860
|
||||
# run, -u to not buffer stdout / stderr
|
||||
CMD /docker/mount.sh && \
|
||||
python3 -u scripts/webui.py --outdir /output --ckpt /data/StableDiffusion/model.ckpt ${CLI_ARGS}
|
||||
# sed -i -- 's/8501/7860/g' .streamlit/config.toml && STREAMLIT_SERVER_HEADLESS=true python -u -m streamlit run scripts/webui_streamlit.py --theme.base dark
|
||||
@@ -9,7 +9,5 @@ file.write_text(
|
||||
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)\
|
||||
.replace('img2img_cfg = gr.Slider(minimum=1.0, maximum=30.0', 'img2img_cfg = gr.Slider(minimum=1.0, maximum=60.0')
|
||||
|
||||
)
|
||||
""", 1)
|
||||
)
|
||||
28
services/hlky/mount.sh
Executable file
28
services/hlky/mount.sh
Executable file
@@ -0,0 +1,28 @@
|
||||
#!/bin/bash
|
||||
|
||||
set -Eeuo pipefail
|
||||
|
||||
declare -A MOUNTS
|
||||
|
||||
ROOT=/stable-diffusion/src
|
||||
|
||||
# cache
|
||||
MOUNTS["/root/.cache"]=/data/.cache
|
||||
# ui specific
|
||||
MOUNTS["${ROOT}/gfpgan/experiments/pretrained_models/GFPGANv1.3.pth"]=/data/GFPGAN/GFPGANv1.4.pth
|
||||
MOUNTS["${ROOT}/realesrgan/experiments/pretrained_models"]=/data/RealESRGAN
|
||||
MOUNTS["${ROOT}/latent-diffusion/experiments/pretrained_models"]=/data/LDSR
|
||||
# hacks
|
||||
MOUNTS["/stable-diffusion/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
|
||||
8
services/hlky/userconfig_streamlit.yaml
Normal file
8
services/hlky/userconfig_streamlit.yaml
Normal file
@@ -0,0 +1,8 @@
|
||||
general:
|
||||
outdir: /outputs
|
||||
default_model: "Stable Diffusion v1.4"
|
||||
default_model_path: /data/StableDiffusion/model.ckpt
|
||||
outdir_txt2img: /outputs/txt2img-samples
|
||||
outdir_img2img: /outputs/img2img-samples
|
||||
optimized: True
|
||||
optimized_turbo: True
|
||||
59
services/lstein/Dockerfile
Normal file
59
services/lstein/Dockerfile
Normal file
@@ -0,0 +1,59 @@
|
||||
# syntax=docker/dockerfile:1
|
||||
|
||||
FROM continuumio/miniconda3:4.12.0
|
||||
|
||||
SHELL ["/bin/bash", "-ceuxo", "pipefail"]
|
||||
|
||||
ENV DEBIAN_FRONTEND=noninteractive
|
||||
|
||||
# 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 rsync gcc -y && apt-get clean
|
||||
|
||||
|
||||
RUN <<EOF
|
||||
git clone https://github.com/invoke-ai/InvokeAI.git stable-diffusion
|
||||
cd stable-diffusion
|
||||
git reset --hard a1739a73b48bfe98b6abcb67f5a0197a9ad270e0
|
||||
sed -i -- 's/python=3.8.5/python=3.9/g' environment.yaml
|
||||
conda env update --file environment.yaml -n base
|
||||
conda clean -a -y
|
||||
EOF
|
||||
|
||||
|
||||
ARG BRANCH=development SHA=8c9f2ae705cf723d4a8a73c416e8d8bf2d746977
|
||||
# this breaks on generation:
|
||||
# there is a new UI anyway, but it is not by any means ready.
|
||||
# ARG BRANCH=development SHA=bdbc76fcd4bd3362312dc91b087d9af66de423b1
|
||||
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 transformers==4.19.2
|
||||
|
||||
COPY . /docker/
|
||||
RUN <<EOF
|
||||
python3 /docker/info.py /stable-diffusion/static/dream_web/index.html
|
||||
chmod +x /docker/mount.sh
|
||||
sed -i -- 's/outputs\//\/output/g' /stable-diffusion/backend/server.py
|
||||
EOF
|
||||
|
||||
|
||||
ENV TRANSFORMERS_CACHE=/cache/transformers TORCH_HOME=/cache/torch PRELOAD=false CLI_ARGS=""
|
||||
WORKDIR /stable-diffusion
|
||||
EXPOSE 7860
|
||||
|
||||
CMD /docker/mount.sh && \
|
||||
# python3 -u backend/server.py --host 0.0.0.0 --port 7860 --cors http://localhost:7860
|
||||
# python3 -u scripts/dream.py --outdir /output --web --host 0.0.0.0 --port 7860 ${CLI_ARGS}
|
||||
echo The lstein webUI is currently deactivated due to implementation limitations: \
|
||||
https://github.com/invoke-ai/InvokeAI/blob/8c9f2ae705cf723d4a8a73c416e8d8bf2d746977/backend/modules/create_cmd_parser.py#L26 \
|
||||
Once the path the output is fixed, the UI will be activated again
|
||||
10
services/lstein/info.py
Normal file
10
services/lstein/info.py
Normal file
@@ -0,0 +1,10 @@
|
||||
import sys
|
||||
from pathlib import Path
|
||||
|
||||
file = Path(sys.argv[1])
|
||||
file.write_text(
|
||||
file.read_text()\
|
||||
.replace('GitHub site</a>', """
|
||||
GitHub site</a>, Deployed with <a href="https://github.com/AbdBarho/stable-diffusion-webui-docker/">stable-diffusion-webui-docker</a>
|
||||
""", 1)
|
||||
)
|
||||
29
services/lstein/mount.sh
Executable file
29
services/lstein/mount.sh
Executable file
@@ -0,0 +1,29 @@
|
||||
#!/bin/bash
|
||||
|
||||
set -eu
|
||||
|
||||
ROOT=/stable-diffusion
|
||||
|
||||
mkdir -p "${ROOT}/models/ldm/stable-diffusion-v1/"
|
||||
ln -sf /cache/models/model.ckpt "${ROOT}/models/ldm/stable-diffusion-v1/model.ckpt"
|
||||
|
||||
base="${ROOT}/src/gfpgan/experiments/pretrained_models/"
|
||||
mkdir -p "${base}"
|
||||
# TODO: "real" GFPGANv1.4.pth
|
||||
ln -sf /cache/models/GFPGANv1.3.pth "${base}/GFPGANv1.4.pth"
|
||||
echo "Mounted GFPGANv1.3.pth"
|
||||
|
||||
# facexlib
|
||||
FACEX_WEIGHTS=/opt/conda/lib/python3.9/site-packages/facexlib/weights
|
||||
|
||||
rm -rf "${FACEX_WEIGHTS}"
|
||||
mkdir -p /cache/weights
|
||||
ln -sf -T /cache/weights "${FACEX_WEIGHTS}"
|
||||
|
||||
REALESRGAN_WEIGHTS=/opt/conda/lib/python3.9/site-packages/realesrgan/weights
|
||||
rm -rf "${REALESRGAN_WEIGHTS}"
|
||||
ln -sf -T /cache/weights "${REALESRGAN_WEIGHTS}"
|
||||
|
||||
if "${PRELOAD}" == "true"; then
|
||||
python3 -u scripts/preload_models.py
|
||||
fi
|
||||
Reference in New Issue
Block a user