Crate diffusers

source ·
Expand description

Diffusion pipelines and models

This is a Rust port of Hugging Face’s diffusers Python api using Torch via the tch-rs.

This library includes:

  • Multiple type of UNet based models, with a ResNet backend.
  • Training examples including version 1.5 of Stable Diffusion.
  • Some basic transformers implementation for handling user prompts.

The models can used pre-trained weights adapted from the Python implementation.