An experimental Stable Diffusion frontend
Project description
Osmosis
An experimental Stable Diffusion web frontend.
Warning This project is not production ready! For a more actively developed and advanced project, check out InvokeAI :p
Features
- Diffusers models
- CoreML models (macOS)
- Checkpoint / Safetensors file models
- Half-precision (
fp16
) support for main models - Diffusers to CoreML model conversion
- Real-ESRGAN upscaling
- GFPGAN face restoration
- Structured metadata according to https://github.com/invoke-ai/InvokeAI/issues/266
- Gallery view
- Image to image
- Inpainting / outpainting
- Training with LoRA / Textual Inversion / Dreambooth
- xFormers optimization
- Apple Silicon optimization
System Requirements
- Ideally, a NVIDIA or Apple Silicon GPU
- At least 10 GB of RAM
- As much disk space as your models require
- A modern browser
Installation
Osmosis is a web app distributed as a package on PyPI.
Ideally, use pipx to install Osmosis in its own isolated environment. First install pipx, then run
$ pipx install osmosis
to install. Alternatively, simple use pip
to install globally:
$ pip install [--user] osmosis
NVIDIA instructions
If you're on a NVIDIA GPU, replace osmosis
in the install scripts with osmosis[xformers]
to enable xFormers optimization support. In addition, add --pip-args "--extra-index-url https://download.pytorch.org/whl/cu117"
to the end to install a CUDA-enabled PyTorch build.
$ pipx install 'osmosis[xformers]' --pip-args "--extra-index-url https://download.pytorch.org/whl/cu117"
# or in pip:
$ pip install [--user] 'osmosis[xformers]' --extra-index-url https://download.pytorch.org/whl/cu117
Linux / AMD instructions
If you're on Linux and using an AMD GPU, you can also use ROCm-optimized PyTorch wheels on install.
$ pipx install osmosis --pip-args "--extra-index-url https://download.pytorch.org/whl/rocm5.2"
# or in pip:
$ pip install [--user] osmosis --extra-index-url https://download.pytorch.org/whl/rocm5.2
Running
$ osmosis
is all you need to start the web UI.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.