查看“︁Stable Diffusion”︁的源代码


= Dockerfile=

<syntaxhighlight lang="dockerfile">
FROM nvidia/cuda:12.4.1-cudnn-devel-ubuntu22.04

RUN sed -i 's|http://ports.ubuntu.com/|https://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/|g' /etc/apt/sources.list /etc/apt/sources.list && \
    apt update && \
    apt install -y build-essential cmake wget git python3 python3-pip sudo
RUN pip3 install jupyterlab
RUN apt-get install -y openssh-client openssh-server && \
    mkdir -p /var/run/sshd && \
    sed -ri 's/^PermitRootLogin\s+.*/PermitRootLogin yes/' /etc/ssh/sshd_config && \
    cat /etc/ssh/ssh_config | grep -v StrictHostKeyChecking > /etc/ssh/ssh_config.new && \
    echo "    StrictHostKeyChecking no" >> /etc/ssh/ssh_config.new && \
    mv /etc/ssh/ssh_config.new /etc/ssh/ssh_config

CMD ["sleep", "infinity"]  
</syntaxhighlight>

Check GPU:
<syntaxhighlight lang="bash">
nvidia-smi
Tue May  7 11:47:20 2024
+---------------------------------------------------------------------------------------+
| NVIDIA-SMI 535.54.03              Driver Version: 535.54.03    CUDA Version: 12.2     |
|-----------------------------------------+----------------------+----------------------+
| GPU  Name                 Persistence-M | Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp   Perf          Pwr:Usage/Cap |         Memory-Usage | GPU-Util  Compute M. |
|                                         |                      |               MIG M. |
|=========================================+======================+======================|
|   0  NVIDIA GeForce RTX 2080 Ti     On  | 00000000:85:00.0 Off |                  N/A |
| 36%   28C    P8              20W / 250W |   2847MiB / 11264MiB |      0%      Default |
|                                         |                      |                  N/A |
+-----------------------------------------+----------------------+----------------------+

+---------------------------------------------------------------------------------------+
| Processes:                                                                            |
|  GPU   GI   CI        PID   Type   Process name                            GPU Memory |
|        ID   ID                                                             Usage      |
|=======================================================================================|
+---------------------------------------------------------------------------------------+
</syntaxhighlight>

= Stable Diffusion =
== Webui==

<syntaxhighlight lang="bash">
sudo apt install wget git python3 python3-venv libgl1 libglib2.0-0
git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
cd stable-diffusion-webui
bash webui.sh
</syntaxhighlight>

To change configuration, edit webui-user.sh:
<syntaxhighlight lang="bash">
export COMMANDLINE_ARGS="--listen --port 5050 --subpath=/kas/py36lph9lctmsk11hp7gngmbt37n"
</syntaxhighlight>

To use new models, download *.safetensors and put into /models/xx folder.

== ComfyUI==

<syntaxhighlight lang="bash">
ls ComfyUI/models/checkpoints
-rw-rw-r-- 1 kas kas 6.5G May  7 13:55 sd_xl_base_1.0.safetensors
-rw-rw-r-- 1 kas kas 6.5G May  7 13:57 sd_xl_base_1.0_0.9vae.safetensors
-rw-rw-r-- 1 kas kas 5.7G Jul 26  2023 sd_xl_refiner_1.0.safetensors
-rw-rw-r-- 1 kas kas 5.7G Jul 29  2023 sd_xl_refiner_1.0_0.9vae.safetensors
-rw-rw-r-- 1 kas kas 4.0G May  7 13:58 v1-5-pruned-emaonly.safetensors
-rw-rw-r-- 1 kas kas 4.9G May  7 13:59 v2-1_768-ema-pruned.safetensors

python3 main.py --port 5050 --listen
</syntaxhighlight>

Use *vae.safetensors<ref>https://www.tinydesk.ai/post/what-is-sdxl-vae</ref>.

[[Category:Linux/Unix]]
[[Category:AI]]