Add stochastic sampling to DPM solver (SDE)

finegrain-ai / refiners

A microframework on top of PyTorch with first-class citizen APIs for foundation model adaptation

https://refine.rs

MIT License

584 stars 43 forks source link

Closed limiteinductive closed 1 month ago

limiteinductive commented 1 month ago

Add the possibility to enable stochastic sampling with the DPM solver. This corresponds to "DPM++ 3M SDE" in k-diffusion repo.