Jalankan Stable Diffusion dengan ComfyUI di Server GPU

Menjalankan pembuatan gambar di server GPU Anda sendiri berarti tidak ada log prompt yang disimpan pihak ketiga, tidak ada filter konten yang tidak Anda pilih, dan biaya yang tetap sama berapa pun jumlah gambar yang Anda buat. Panduan ini menyiapkan ComfyUI, antarmuka berbasis node untuk Stable Diffusion dan model serupa, di Ubuntu.

Pilih GPU

Memori video adalah kendalanya. 8 GB menjalankan SD 1.5 dengan nyaman, 12 sampai 16 GB menangani SDXL dengan baik, dan 24 GB memberi ruang untuk model yang lebih baru dan lebih besar serta resolusi lebih tinggi. Jika ragu, lihat GPU konsumen versus datacenter. Sisakan juga sekitar 50 GB disk, karena file model berukuran besar.

Langkah 1: periksa driver

nvidia-smi

Jika GPU Anda tampil, berarti driver berfungsi. Jika tidak, pasang driver yang direkomendasikan dengan sudo ubuntu-drivers autoinstall lalu reboot.

Langkah 2: pasang ComfyUI

sudo apt update && sudo apt install -y git python3-venv
git clone https://github.com/comfyanonymous/ComfyUI
cd ComfyUI
python3 -m venv venv
. venv/bin/activate
pip install torch torchvision --index-url https://download.pytorch.org/whl/cu121
pip install -r requirements.txt

Sesuaikan indeks wheel CUDA dengan versi yang direkomendasikan PyTorch untuk driver Anda. Periksa halaman instalasi mereka jika perintah di atas sudah berubah.

Langkah 3: tambahkan model

Unduh checkpoint yang lisensinya boleh Anda gunakan ke dalam ComfyUI/models/checkpoints/. Baca lisensi model untuk penggunaan komersial sebelum membangun produk di atasnya.

cd models/checkpoints
wget -O model.safetensors "https://example.com/path/to/model.safetensors"

Langkah 4: jalankan secara privat

Secara bawaan ComfyUI hanya mendengarkan di localhost, dan itulah yang Anda inginkan. Aksesnya melalui terowongan SSH.

python main.py --listen 127.0.0.1 --port 8188

ssh -L 8188:127.0.0.1:8188 user@your-server

Buka http://127.0.0.1:8188 di komputer Anda sendiri. Jangan buka port ke internet: secara bawaan tidak ada login.

Langkah 5: biarkan tetap berjalan

sudo tee /etc/systemd/system/comfyui.service > /dev/null <<'UNIT'
[Unit]
Description=ComfyUI
After=network-online.target

[Service]
User=ubuntu
WorkingDirectory=/home/ubuntu/ComfyUI
ExecStart=/home/ubuntu/ComfyUI/venv/bin/python main.py --listen 127.0.0.1
Restart=on-failure

[Install]
WantedBy=multi-user.target
UNIT
sudo systemctl daemon-reload && sudo systemctl enable --now comfyui

Sesuaikan pengguna dan path dengan server Anda.

Biaya dan privasi dibandingkan alat yang di-host

• Biaya: GPU per jam harganya sama baik Anda membuat sepuluh gambar maupun sepuluh ribu. Ini lebih murah daripada harga per gambar saat Anda membuat gambar dalam jumlah besar, dan lebih mahal saat jarang membuat, jadi hentikan server saat menganggur.
• Privasi: prompt dan hasil tetap berada di mesin yang Anda kendalikan, dan Anda bisa membayar dengan Monero atau koin lain tanpa ID.
• Kendali: Anda memilih model, sampler, dan pengaturan keamanan, dan Anda bertanggung jawab atas cara menggunakannya.

Untuk model teks pada perangkat keras yang sama, lihat Ollama dan Open WebUI.