Skip to content

A Unity package to run pretrained diffusion models with Unity Sentis

License

Notifications You must be signed in to change notification settings

julienkay/com.doji.diffusers

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

🚧 Experimental! – This package is in an early, exploratory stage and not production-ready. Expect sharp edges, breaking changes, and general chaos.

doji logo

Diffusers

A Unity package to run pretrained diffusion models with Unity Sentis

OpenUPM · Documentation (coming soon) · Feedback/Questions

About

This is essentially a port of Hugging Face’s diffusers library.

As of today only a limited number of pipelines and schedulers are supported (see below).

Installation

via OpenUPM
  1. In Edit -> Project Settings -> Package Manager, add a new scoped registry:

    Name: Doji
    URL: https://package.openupm.com
    Scope(s): com.doji
    
  2. In the Package Manager install com.doji.diffusers either by name or select it in the list under Package Manager -> My Registries

  3. For the time being, you also have to use a custom fork of Sentis: In the package Manager -> Package Manager -> Install package from git URL-> https://github.com/julienkay/com.unity.sentis.git

Features/Roadmap

Some things that might be worked on next are:

  • add/fix classifier-free guidance
  • support more models
    • SD 1.5
    • SD 2.1
    • SDXL
    • SD-Turbo
    • SDXL-Turbo
    • LoRA models
  • more scheduler implementations
    • PNDM
    • DDIM
    • EulerDiscrete
    • EulerAncestralDiscrete
    • LCM
    • DDPM
    • KDPM2Discrete
    • KDPM2AncestralDiscrete
  • more pipeline/model types
    • txt2img
    • img2img,
    • inpaint,
    • upscale,
    • depth2img
    • marigold
  • async pipeline methods (sliced inference)
  • support multiple images per prompt
  • write some documentation

Dependencies