LaurentMazare / diffusers-rs

An implementation of the diffusers api in Rust
Apache License 2.0
521 stars 54 forks source link

Adapt pipeline examples and scheduler's to huggingface diffusers interface #35

Closed mspronesti closed 1 year ago

mspronesti commented 1 year ago

This PR aims at generalizing the three pipeline examples to make them work with the schedulers that need to scale the denoising model input depending on the current timestep and that have a initial noise standard deviation that depends on the timesteps.

This is done making all of them implementing the following methods

The pipeline examples are modified as follows:

LaurentMazare commented 1 year ago

Looks good both for the scheduler itself and for the enhanced api, thanks!