JuliaNeuralGraphics / Diffusers.jl

Diffusion models for image generation in Julia

Diffusers.jl

An implementation of the subset of Hugging Face Diffusers in pure Julia. Provides text-to-image StableDiffusion implementation.

Painting of a farmer in the field

Painting of a Dome Da Vinchi	Painting of a Dome Da Vinchi	Painting of a Dome Van Gogh	Painting of a Dome Van Gogh

Refractive spheres sunlight	Refractive spheres	Thunder in mountains dark clouds	Edo era computer

Installation

1. Clone the repo.
2. Launch Julia REPL from Diffusers.jl directory:

julia --threads=auto --project=.

3. Instantiate & update with ]up command.

Usage

julia> using AMDGPU # For AMD GPU support
julia> using CUDA   # For Nvidia GPU support

julia> using Flux
julia> Flux.gpu_backend!("AMDGPU") # For AMD GPU support
julia> Flux.gpu_backend!("CUDA") # For Nvidia GPU support

julia> using Diffusers
julia> Diffusers.main(["painting of a farmer in the field"]; device=gpu, precision=f16)

Images are saved in the Diffusers.jl directory.