Skip to content

Latest commit

 

History

History
41 lines (30 loc) · 2.21 KB

README.md

File metadata and controls

41 lines (30 loc) · 2.21 KB
Raw

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-farmer-in-the-field-2
Painting of a Dome Da Vinchi Painting of a Dome Da Vinchi Painting of a Dome Van Gogh Painting of a Dome Van Gogh
painting-of-a-dome-leonardo-da-vinchi-2 painting-of-a-dome-leonardo-da-vinchi-1 painting-of-a-dome-van-gogh-2 painting-of-a-dome-van-gogh-2 (copy)
Refractive spheres sunlight Refractive spheres Thunder in mountains dark clouds Edo era computer
refractive-spheres-sunlight-1 refractive-spheres-1 thunder-in-mountains-dark-clouds-1 edo-era-computer-1

Installation

  1. Clone the repo.
  2. Launch Julia REPL from Diffusers.jl directory:
julia --threads=auto --project=.
  1. 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.