/docker-stable-diffusion-2

Text to Image AI Generation using Stable Diffusion 2

Primary LanguageJavaScriptMIT LicenseMIT

Docker Stable Diffusion 2

A fork from dalle-playground using Stable Diffusion 2 for CPU

AboutUsageCreditsLicense

screenshot

About

This repository is a fork from dalle-playground that uses Stable Diffusion 2 in CPU only.

Usage

Run in your terminal docker compose up, wait the install and you can access from your browser Website on http://localhost:3000/. On the website put this on backend url http://127.0.0.1:8080/. To shutdown the application run docker compose down.

Manual install

python -m venv venv

. venv/bin/activate

pip install -U pip

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu

pip install "jax[cpu]"

pip install diffusers transformers accelerate scipy safetensors

pip install Flask

pip install flask-cors

pip install flask-restful

Credits

Thanks for the open source projects bellow:

License

MIT