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.
Modules§
- models
- Models
- pipelines
- Pipelines
- schedulers
- Diffusion pipelines and models
- transformers
- Transformers
- utils